我尝试在install_requires中指定PyTorch和CUDA。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中执行此操作？
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
gpfsuwkq1#
我也面临着同样的问题，后来我修复了它，但使用这个在setup.py文件和它的工作，只是添加这些行在您的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获得你所需版本的链接
1条答案
按热度按时间gpfsuwkq1#
我也面临着同样的问题，后来我修复了它，但使用这个在
setup.py文件和它的工作，只是添加这些行在您的setup.py文件。所有这些都是针对linux版本的，如果你想要任何MACOSX或Windows，只要在每行@后更改链接，你就可以从https://download.pytorch.org/whl/torch_stable.html获得你所需版本的链接