在 setup.py 中使用 CUDA 安装 PyTorch

Question

我正在尝试在 install_requires 中指定带有 CUDA 的 PyTorch。使用 pip 安装的命令是

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

如何在 setup.py install_requires 中做到这一点？

Answer 1

后来我也遇到了同样的问题，我修复了它，但是在文件中使用它setup.py并且它有效，只需在setup.py文件中添加这些行即可。

"torch@https://download.pytorch.org/whl/cu111/torch-1.8.0%2Bcu111-cp37-cp37m-linux_x86_64.whl",
"torchvision@https://download.pytorch.org/whl/cu111/torchvision-0.9.0%2Bcu111-cp37-cp37m-linux_x86_64.wh",
"torchaudio@https://download.pytorch.org/whl/torchaudio-0.8.0-cp36-cp36m-linux_x86_64.whl"

所有这些都是针对 Linux 版本的，如果您想要任何 MACOSX 或 Windows，只需更改每行中 @ 之后的链接，您就可以从 https://download.pytorch.org/whl/torch_stable.html获取所需版本的链接