如何在 Visual Studio 中使用 pip 安装支持 CUDA 的 pytorch

Question

torch with CUDA enabled我正在尝试在 Visual Studio 环境中安装。我右键单击“解决方案资源管理器”中的“Python 环境”，卸载了未使用 CUDA 编译的现有 Torch 版本，并尝试pip command从官方Pytorch网站运行它。命令是：

pip3 install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

Visual Studio 报告此错误Looking in links: https://download.pytorch.org/whl/cu102/torch_stable.html ERROR: Could not find a version that satisfies the requirement pip3 (from versions: none) ERROR: No matching distribution found for pip3。

我在这个网站上看到过类似的问题，但有些是回避的，Conda而另一些则确实有不明确的答案，但没有被接受，所以我怀疑是否要遵循答案。我有一个非常重要的项目需要展示，除非我安装启用了 cuda 的 torch，否则我无法做到这一点，请帮助我，谢谢。

Answer 1

我按照pytorch 网站上的主要安装说明进行操作，但仍然未能成功。对于我的设置，这导致了pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117.

我不得不添加--upgrade --force-reinstall最终解决了问题。

Answer 2

你可以在pytorch以前版本的网站上查看。nvcc --version首先，使用以下命令确保您的计算机中有 cuda

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