问题及解决方案
可能是cudnn出现问题了。可以根据这篇文章的命令验证一下torch.backends.cudnn.version()
解决方案
参考知乎https://zhuanlan.zhihu.com/p/333362435,第3.4节和第4节的的教程进行操作,解压cudnn文件后vim ~/.bashrc或者vim ~/.zshrc修改路径
export CUDA_HOME=/xxx/cudnn-linux-xxx#改成自己的解压路径
export LD_LIBRARY_PATH=${CUDA_HOME}/lib:$LD_LIBRARY_PATH