问Pytorch 1.4支持cuda10.1吗？
EN

Stack Overflow用户

提问于 2020-05-20 11:23:43

回答 1查看 807关注 0票数 0

我试着安装pytorch 1.5+cu101，但是它找不到caffe2_nvrtc.dll(它确实存在于lib\caffe2_nvrtc.dll中)。所以我试着安装pytorch 1.4+cu101，但是我在https://download.pytorch.org/whl/torch_stable.html上找不到这样的轮子(只有pytorch1.4+cu100/cu92)。

有什么帮助吗？

python

pytorch

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-05-20 14:29:47

默认情况下，PyTorch 1.4.0随CUDA10.1一起提供，因此没有单独的带有cu101后缀的包，这些包只适用于其他版本。您只需要安装常规的torch包：

pip install torch==1.4.0 -f https://download.pytorch.org/whl/torch_stable.html

票数 1

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

原文链接：

https://stackoverflow.com/questions/61911837

复制

相似问题

问Pytorch 1.4支持cuda10.1吗？
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Pytorch 1.4支持cuda10.1吗？EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Pytorch 1.4支持cuda10.1吗？
EN