由于您没有提供操作系统的信息,我无法提供特定于操作系统的命令。但是,下面是在 Windows、Linux 或 macOS 上下载 PyTorch 1.8 GPU 版本的命令。

在 Windows 上下载 PyTorch 1.8 GPU 版本:

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

在 Linux 或 macOS 上下载 PyTorch 1.8 GPU 版本:

pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

请注意,此命令假定您已经安装了适当的 CUDA 驱动程序和 cuDNN 库。如果您的系统中没有这些库,请先安装它们,然后再运行上述命令。

PyTorch 1.8 GPU 版本下载命令 - Windows、Linux 和 macOS

原文地址: http://www.cveoy.top/t/topic/lHqO 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录