一、安装nvidia驱动
1、查看显卡型号
lspci | grep -i vga
2、官网上查看对应型号的驱动
https://www.nvidia.cn/drivers/lookup/
搜一下型号
选个驱动
3、安装驱动，一路ok/yes的就完了
sh NVIDIA-Linux-x86_64-550.76.run --no-opengl-files
4、检查一下，成没成
nvidia-smi
有这个就是行了
5、这是驱动装错了，换一个
换一个驱动
二、安装docker-nvidia
1、打开终端，并执行以下命令来添加NVIDIA Docker的存储库：
添加NVIDIA Docker存储库密钥
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
获取NVIDIA Docker存储库密钥
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
2、安装NVIDIA Docker 2.0
sudo apt-get update && sudo apt-get install -y nvidia-docker2
3、重启
sudo systemctl restart docker
4、验证
docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
