在 setup.py 中使用 CUDA 安装 PyTorch

【问题标题】：Installing PyTorch with CUDA in setup.py在 setup.py 中使用 CUDA 安装 PyTorch
【发布时间】：2021-06-18 15:37:00
【问题描述】：

我正在尝试在 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 中做到这一点？

【问题讨论】：

可以指定轮文件stackoverflow.com/a/59122860/6210807

标签： python pytorch setup.py install-requires

【解决方案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"

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

【讨论】：