查看GPU型号
lspci | grep -i nvidia

查看NVIDIA驱动版本
sudo dpkg --list | grep nvidia-*

nvidia-smi是用来查看GPU版本信息，GPU使用信息查询：
nvidia-smi

驱动下载

• https://www.nvidia.com/Download/Find.aspx?spm=a2c4g.11186623.0.0.2d45573b4oPoJ0&lang=cn

#查找芯片对应的版本

nvidia-container-toolkit 安装

• https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)  
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -  
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list  

apt-get update 
apt-get install -y nvidia-container-toolkit

tensorflow测试GPU

python
import tensorflow as tf
print('GPU',tf.test.is_gpu_available())
a = tf.constant(2.)
b = tf.constant(4.)

print(a * b)

#如果是true就能调用GPU，如果false就还是tf和cuda版本的问题

cuda降版本方法

卸载cuda
sudo apt-get --purge remove "cublas" "cuda*"
卸载驱动
sudo apt-get --purge remove "nvidia"

#报错解决方法
#保留高版本的驱动，安装低版本的cuda
add-apt-repository ppa:graphics-drivers/ppa
apt search nvidia
cuda10.2回退到10.1，run文件中驱动相应的由440->418（或430）