我正试图在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中做到这一点？
我后来也遇到了同样的问题，我修复了它，但在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