Nvidia docker安装教程

一、更新显卡驱动到410.78版本

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-410
nvidia-smi

二、docker安装

1.卸载旧版本docker
sudo apt-get remove docker docker-engine docker.io

2.添加CA证书
sudo apt-get update
sudo apt-get install \
    apt-transport-https \
    ca-certificates \
    curl \
    software-properties-common

3.添加
curl -fsSL https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu/gpg | sudo apt-key add -

4.添加docker源
sudo add-apt-repository \
"deb [arch=amd64] https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu \
$(lsb_release -cs) \
stable"

5.安装docker-ce
sudo apt-get update
sudo apt-get install docker-ce

6.启动docker
sudo systemctl enable docker
sudo systemctl start docker

7.测试docker是否安装成功
sudo docker run hello-world

---

三、nvidia-docker 安装

1.添加源 https://blog.csdn.net/weixin_39610043/article/details/85334877
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \ sudo apt-key add - curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64/nvidia-docker.list | \ sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update

2.安装nvidia-docker2软件包
sudo apt-get install nvidia-docker2

如果提示类似下面错误：
The following packages have unmet dependencies:
 nvidia-docker2 : Depends: docker-ce (= 17.06.2~ce-0~ubuntu) but 18.06.3~ce~3-0~ubuntu is to be installed or
                           docker-ee (= 17.06.2~ee-0~ubuntu) but it is not installable
E: Unable to correct problems, you have held broken packages.

查看docker支持版本
apt-cache madison docker-ce

安装docker低版本
sudo apt-get install docker-ce=17.06.2~ce-0~ubuntu

再重新安装指定版本nvidia-docker2
apt-get install nvidia-docker2=2.0.3+docker17.06.2-1 nvidia-container-runtime=2.0.0+docker17.06.2-1 -y

3.重新加载Docker守护程序配置
sudo pkill -SIGHUP dockerd

4.测试
sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi