文章/答案/技术大牛

发布

社区首页 >问答首页 >在setup.py中安装带有CUDA的PyTorch

问在setup.py中安装带有CUDA的PyTorch
EN

Stack Overflow用户

提问于 2021-03-22 07:03:14

回答 1查看 361关注 0票数 2

我正在尝试在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中执行此操作？

install-requires

python

pytorch

setup.py

回答 1

Stack Overflow用户

发布于 2021-10-07 17:49:27

我后来也遇到了同样的问题，后来我修复了它，但在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得到你想要的版本的链接。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/66738473

复制

相似问题

问在setup.py中安装带有CUDA的PyTorch
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问在setup.py中安装带有CUDA的PyTorchEN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问在setup.py中安装带有CUDA的PyTorch
EN