Ubuntu 18.04 配置 nvidia-docker

准备 docker

1. 使用脚本自动安装 docker

$ curl -fsSL get.docker.com -o get-docker.sh
$ sudo sh get-docker.sh --mirror Aliyun
# $ sudo sh get-docker.sh --mirror AzureChinaCloud

2. 使用 apt 自动安装 docker

$ sudo apt install docker

3. 手动安装 docker

卸载旧版本 docker

$ sudo apt-get remove docker docker-engine docker.io

使用 apt 手动安装

由于 apt 源使用 HTTPS 以确保软件下载过程中不被篡改。因此，我们首先需要添加使用 HTTPS 传输的软件包以及 CA 证书。

安装所需的工具包。

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

添加 docker 源

添加软件源（国内）的 GPG 密钥。官方源参见注释。

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

# 官方源
# $ curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

向 sources.list 中添加 Docker 软件源

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


# 官方源
# $ sudo add-apt-repository \
#    "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
#    $(lsb_release -cs) \
#    stable"

安装 docker-ce

$ sudo apt-get update
$ sudo apt-get install docker-ce

将当前用于加入 docker 用户组

默认情况下，docker 命令会使用 Unix socket 与 Docker 引擎通讯。而只有 root 用户和 docker 组的用户才可以访问 Docker 引擎的 Unix socket。出于安全考虑，一般 Linux 系统上不会直接使用 root 用户。因此，更好地做法是将需要使用 docker 的用户加入 docker 用户组。

简单的说，如果不把当前用户加入 docker 用户组，那么每次执行 docker 命令都需要在前面加上 sudo 才可执行命令。

$ sudo usermod -aG docker $USER

4. 启动 docker-CE

$ sudo systemctl enable docker
$ sudo systemctl start docker

5. 检查 docker 是否安装成功

$ docker run hello-world

Unable to find image 'hello-world:latest' locally
latest: Pulling from library/hello-world
d1725b59e92d: Pull complete
Digest: sha256:0add3ace90ecb4adbf7777e9aacf18357296e799f81cabc9fde470971e499788
Status: Downloaded newer image for hello-world:latest

Hello from Docker!
This message shows that your installation appears to be working correctly.

To generate this message, Docker took the following steps:
 1. The Docker client contacted the Docker daemon.
 2. The Docker daemon pulled the "hello-world" image from the Docker Hub.
    (amd64)
 3. The Docker daemon created a new container from that image which runs the
    executable that produces the output you are currently reading.
 4. The Docker daemon streamed that output to the Docker client, which sent it
    to your terminal.

To try something more ambitious, you can run an Ubuntu container with:
 $ docker run -it ubuntu bash

Share images, automate workflows, and more with a free Docker ID:
 https://hub.docker.com/

For more examples and ideas, visit:
 https://docs.docker.com/get-started/

准备 nvidia 驱动

安装 nvidia-docker

在使用带有 cuda 环境的 docker 容器之前，首先需要安装 nvidia-docker 组件

添加 nvidia-docker 源

$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update

安装 nvidia-docker2

安装 nvidia-docker2 后重启 docker 使得 nvidia-docker2 生效。

$ sudo apt-get install nvidia-docker2
$ sudo systemctl restart docker

参考

https://www.cnblogs.com/pprp/p/9430836.html
http://www.ityouknow.com/docker/2018/03/12/docker-use-dockerfile.html
https://yeasy.gitbook.io/docker_practice/install/ubuntu
https://blog.csdn.net/BigData_Mining/article/details/99681168