NVIDIA Docker v2 安裝

2018-03-02

技術Container

Docker, NVIDIA

2017 年 11 月 NVIDIA 已將 NVIDIA Docker v2 的版本合併(merged)至 NVIDIA/nvidia-docker 的 repository，這意味著 v2 會逐漸取代 v1。

而根據官方的說明，v1 與 v2 差異如下：

不需要封裝的 Docker CLI 以及獨立的背景程式(daemon)
GPU 的隔離現在透過環境變數NVIDIA_VISIBLE_DEVICES
任何的 Docker image 都可以啟動 GPU 支援，不只是基於官方的 CUDA image
Ubuntu 與 CentOS 都有 Package repositories，意味著安裝更為簡單與方便
基於libnvidia-container來重新實現

而本文將介紹如何使用 NVIDIA Docker v2 來讓容器使用 GPU。

安裝環境

環境資訊：

Name	Spec
OS	Ubunut 16.04 LTS
CPU	4 CPU
RAM	16 GB
Disk	1 TB

事情準備

安裝前需要確認以下幾點：

GNU/Linux x86_64 with kernel version > 3.10
Docker >= 1.12
NVIDIA GPU with Architecture > Fermi (2.1)
NVIDIA drivers ~= 361.93 (untested on older versions)

Docker 版本要安裝 stable release 的版本，請參考這裡。

安裝 Docker CE

首先，需要在環境中安裝Docker CE，若使用 Docker 官方提供的快速安裝腳本的話，安裝的Docker CE版本並非 stable 的版本。所以 nvidia-docker 可能並未支援。因此，若要安裝 stable 的 Docker CE 請參考 Docker CE 安裝，這裡不再詳述如何安裝 Docker CE。

安裝 nvidia-docker

Docker CE 安裝完成後，接著可以開始安裝nvidia-docker2：

$ apt-get install nvidia-384
$ docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
$ apt-get purge -y nvidia-docker
$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -
$ curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64/nvidia-docker.list | \
    tee /etc/apt/sources.list.d/nvidia-docker.list
$ apt-get update
$ apt-get install -y nvidia-docker2
$ pkill -SIGHUP dockerd

安裝完成後，使用nvidia-docker指令確認是否安裝成功：

nvidia-docker version
NVIDIA Docker: 2.0.3
Client:
 Version:      18.03.1-ce
 API version:  1.37
 Go version:   go1.9.5
 Git commit:   9ee9f40
 Built:        Thu Apr 26 07:17:20 2018
 OS/Arch:      linux/amd64
 Experimental: false
 Orchestrator: swarm

Server:
 Engine:
  Version:      18.03.1-ce
  API version:  1.37 (minimum version 1.12)
  Go version:   go1.9.5
  Git commit:   9ee9f40
  Built:        Thu Apr 26 07:15:30 2018
  OS/Arch:      linux/amd64
  Experimental: false

驗證 nvidia-docker

下載官方的 Docker image，並使用nvidia-smi指令來驗證在 Docker Container 中有抓到 GPU 資訊：

$ nvidia-docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
Tue Mar  2 06:19:09 2018
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.111                Driver Version: 384.111                   |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 106...  Off  | 00000000:01:00.0 Off |                  N/A |
|  0%   35C    P0    15W / 120W |      0MiB /  3013MiB |      2%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

安裝環境

事情準備

安裝 Docker CE

安裝 nvidia-docker

驗證 nvidia-docker

參考資料