首先确保NVIDIA Drivers和Docker已经安装好；

其中NVIDIA Drivers安装可参考链接：linux64/ubuntu20.04安装NVIDIA驱动详细过程_Alexand008的博客-CSDN博客

其中Docker安装可参考链接Docker CE的安装：[笔记] Ubuntu 18.04安装Docker CE及NVIDIA Container Toolkit流程 - ET民工[源自火星] - 博客园

备注：snap方式的安装，不确定行不行得通

---------------------------------分割线--------------------------------------------

本地环境是：NVIDIA-SMI 515.48.07    Driver Version: 515.48.07    CUDA Version: 11.7

sudo systemctl start docker && sudo systemctl enable 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
#安装软件包：
sudo apt-get update

此时可能出现bug：
 Unable to fetch some archives, maybe run apt-get update or try with --fix-missing?
解决方案：
sudo apt-get clean 
sudo apt-get update

#安装软件包nvidia-docker2
4、sudo apt-get install -y nvidia-docker2

#重启docker服务：
5、sudo systemctl restart docker

#用官方container测试是否安装成功
 6、sudo nvidia-docker run -e NVIDIA_VISIBLE_DEVICES=all --rm nvidia/cuda:11.0-base nvidia-smi
此时可能有问题，尝试cuda10.0试下；
 sudo nvidia-docker run -e NVIDIA_VISIBLE_DEVICES=all --rm nvidia/cuda:10.0-base nvidia-smi
-----------------------------------------

如果成功的话，会显示如下:

有问题随时交流，欢迎随时交流~

 参考链接：

[笔记] Ubuntu 18.04安装Docker CE及NVIDIA Container Toolkit流程 - ET民工[源自火星] - 博客园

docker学习笔记（9）：nvidia-docker安装、部署与使用_submarineas的博客-CSDN博客_nvidia-docker

彩蛋 

更多干货内容请关注公众号SLearningAI，欢迎CSDN与公众号同步交流^_^