看了下原因为没有装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 
文章出处登录后可见！
已经登录？立即刷新
