当系统有 CUDA 11.2 时我们可以安装 Pytorch CUDA 11.3

Question

我的 PC 中有 cuda 11.2，想要安装 PyTorch。PyTorch 在其网站中仅提及 CUDA10.2 和 11.3

我可以在我的电脑上安装 torch==1.10.1+cu113 吗？如果没有，如何安装 CUDA11.2 的 PyTorch

我不想更改我的 CUDA 版本，因为我有其他应用程序使用它。

Answer 1

我尝试了 11.3 版本，到目前为止它在 GPU 上运行良好：

sudo pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

Answer 2

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

这有效，但不是最新版本