Ubantu 下 Docker 在线安装 GPU

1. 配置apt允许以 HTTPS方式拉取镜像安装

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

2. 添加Docker官方 GPG key

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

3. 更新apt包索引

sudo apt-get update

4. 列出可以安装的Docker版本

apt-cache madison docker-ce

5. 安装17.03.2版本的Docker

sudo apt-get install docker-ce=17.03.2~ce-0~ubuntu-xenial

6. 查看可以安装的nvidia-docker

sudo apt-cache madison nvidia-docker

7. 安装合适的nvidia-docker版本

sudo apt-get install -y nvidia-docker=1.0.1-1

8. 创建daemon.json文件

sudo vim /etc/docker/daemon.json

{
  "dns": ["114.114.114.114"],
  "registry-mirrors": ["https://usdn7gu8.mirror.aliyuncs.com"],
  "insecure-registries": ["registry.cn-hangzhou.aliyuncs.com","cq.xuduan.tech:60419"],
  "default-runtime": "nvidia",
   "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": []
        }
   }
}

9. 启动docker服务

sudo systemctl start docker

10. 启动nvidia-docker服务

sudo systemctl start nvidia-docker

11. 验证nvidia-docker是否可用

sudo docker run --runtime=nvidia --rm nvidia/cuda:8.0-runtime nvidia-smi