AttributeError: module ‘torch‘ has no attribute ‘cuda‘

看了下原因为没有装pytorch。(印象中是装了的不知道什么时候这台服务器没有了。。)

解决方案:到pytorch官网上找到对应的cuda版本的pytorch安装即可

Previous PyTorch Versions | PyTorch

比如我的是cuda 10.2(使用nvcc -V命令查看)

那么就是使用以下命令安装——

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch

 

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
上一篇 2023年9月13日 上午8:14
下一篇 2023年9月13日 上午8:15

相关推荐