Ubuntu16.04安裝Docker、nvidia-docker

Docker安裝

1.更換國內軟體源，推薦中國科技大學的源，穩定速度快（可選）

sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak
sudo sed -i 's/archive.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
sudo apt update

2.安裝需要的包

sudo apt install apt-transport-https ca-certificates software-properties-common curl

3.添加 GPG 密鑰，並添加 Docker-ce 軟體源，這裡還是以中國科技大學的 Docker-ce 源為例

curl -fsSL https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
sudo add-apt-repository "deb [arch=amd64] https://mirrors.ustc.edu.cn/docker-ce/linux/ubuntu \
$(lsb_release -cs) stable"

4.添加成功後更新軟體包緩存

sudo apt update

5.安裝 Docker-ce

sudo apt install docker-ce

6.設置開機自啟動並啟動 Docker-ce（安裝成功後預設已設置並啟動，可忽略）

sudo systemctl enable docker
sudo systemctl start docker

7.測試運行

sudo docker run hello-world

8.添加當前用戶到 docker 用戶組，可以不用 sudo 運行 docker（可選）

sudo groupadd docker
sudo usermod -aG docker $USER

9.測試添加用戶組（可選）

docker run hello-world

Nvidia-docker安裝

1. 卸載 nvidia-docker 1.0 及其他GPU容器

docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker

2. 添加package repositories

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
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
sudo apt-get update

3. 安裝 nvidia-docker2

sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

4. 測試安裝

docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

References

1. https://docs.docker.com/install/

2. https://github.com/NVIDIA/nvidia-docker