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

1、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

2、配置nvidia-docker源
添加源
$ distribution=$(. /etc/os-release;echo I D ID IDVERSION_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