经常遇到装完更新的cuda版本后torch、tf不能用GPU的情况，原因是CUDA和NVIDIA驱动也有对应关系。记录以备忘
CUDA-NVIDIA驱动对应表
一般而言，不同版本的CUDA要求不同的NVIDIA驱动版本,同时显卡驱动版本要不低于CUDA的安装版本，具体的对照关系如下：
如下链接对应了官方的版本要求说明：
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
查看自己的NVIDIA驱动版本:
- linux 下
nvidia-smi
- windows下
桌面右键->nvidia控制面板->左下角系统信息
安装显卡驱动
一般有如下两种方式：
前提需要查询对应直自己机器的显卡驱动版本，查询链接如下：
https://www.nvidia.com/Download/index.aspx?lang=en-us
安装方式如下：
1、apt-get安装
sudo apt-get install nvidia-430
2、下载.run文件离线安装
sudo service lightdm stop
sudo ./NVIDIA-Linux-x86_64-430.14.run
sudo service lightdm start
网友评论