ubuntu16.04安装Docker及NVIDIA Container Toolkit流程

一、ubuntu16.04安装Docker
1.由于apt官方库里的docker版本可能比较旧，所以先卸载可能存在的旧版本
$ sudo apt-get remove docker docker-engine docker-ce docker.io
2.选择国内的云服务商，这里选择阿里云为例
curl -sSL http://acs-public-mirror.oss-cn-hangzhou.aliyuncs.com/docker-engine/internet | sh -
3.安装所需依赖的包
sudo apt-get install linux-image-extra-$(uname -r) linux-image-extra-virtual
4.添加使用HTTPS传输的软件包以及CA证书
sudo apt-get update
sudo apt-get install apt-transport-https ca-certificates
5.添加GPS密码
sudo apt-key adv --keyserver hkp://p80.pool.sks-keyservers.net:80 --recv-keys 58118E89F3A912897C070ADBF76221572C52609D
6.添加软件源
echo “deb https://apt.dockerproject.org/repo ubuntu-xenial main” | sudo tee /etc/apt/sources.list.d/docker.list
7.添加成功后更新软件缓存
sudo apt-get update
8.安装Docker
sudo apt-get install docker-ce
9.启动docker | 设置开机自启动并启动 Docker（安装成功后默认已设置并启动，可忽略）
sudo systemctl enable docker
sudo systemctl start docker
10.测试运行
sudo docker run hello-world
11.添加当前用户到 docker 用户组，可以不用 sudo 运行 docker（可选）
sudo groupadd docker
sudo usermod -aG docker $USER
12.测试添加用户组（可选）
docker run hello-world
13.查看docker版本
docker --version

二、配置nvidia-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

安装并重启docker
$ sudo apt update && sudo apt install -y nvidia-container-toolkit
$ sudo systemctl restart docker

使用
在官方CUDA镜像上测试 nvidia-smi
$ sudo docker run --gpus all nvidia/cuda:9.0-base nvidia-smi

启动支持双GPU的容器
$ sudo docker run --gpus 2 nvidia/cuda:9.0-base nvidia-smi

指定GPU 1，运行容器
$ sudo docker run --gpus device=0 nvidia/cuda:9.0-base nvidia-smi