1.准备工作
驱动版本(Driver Version)、CUDA Toolkit(CUDA Version)、PyTorch版本要相互匹配才能成功使用.
2.查看驱动版本
使用命令查看驱动版本, 如果没有该命令请先安装驱动参考:
CentOS7安装NVIDIA GPU驱动程序和CUDA工具包-CSDN博客
nvidia-smi
可见我当前的驱动版本是520.61,CUDA版本是11.8
安装cuda toolkit 参考:
3.CUDA版本和驱动版本不匹配的应对方法
如果现在你的CUDA版本和驱动版本不匹配,就可以:
- 方法1:使用低版本CUDA Toolkit
- 方法2:更新驱动升级为更高版本