The NVIDIA driver is too old PyTorch version that has been compiled with your version of the CUDA
该方法可检查pytorch与cuda版本是否一致的情况
显示的情况
存在的问题
说明cuda版本与pytorch版本不匹配
解决方法
1.查看torch的版本
python
import torch
torch.__version__
2.查看torch对应的cuda的版本
torch.version.cuda
3.查看cuda是否可用
torch.cuda.is_available()
4.查看cuda的版本
cat /usr/local/cuda/version.txt
5.对比第2步与第4步显示结果是否一致,cuda版本与torch的cuda版本不一致的话
5.1卸载安装的torch: pip uninstall torch
5.2找对应版本的torch 网址: https://pytorch.org/get-started/previous-versions/
安装命令也在网址里,重新安装torch