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