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