解决torch模块中没有cuda属性的AttributeError错误

看了下原因为没有装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

 

物联沃分享整理
物联沃-IOTWORD物联网 » 解决torch模块中没有cuda属性的AttributeError错误

发表评论