Nvidia-Dockerのインストール方法(Ubuntu16.04 LTS)

f:id:AtsuyaKoike:20181101225524p:plain
画像元：https://github.com/NVIDIA/nvidia-docker

WindowsPCにデュアルブートしてUbuntu16.04を使っているのですが、環境を弄ったりするといろいろと面倒くさいのでDockerをインストールしました。今回、その方法をまとめておきます。
PCにNvidiaのグラボが載っているので、Nvidia-Dockerを使います。

参考：
qiita.com

nvidiaドライバのインストール

$ sudo apt update -y
$ sudo apt upgrade -y

場合によっては
$ sudo apt upgrade -y
をやるとgrubの設定画面が出てきますが、ローカルバージョンを保持すれば問題ありません。
f:id:AtsuyaKoike:20181101131430j:plain

Repositoryを追加後、ドライバインストールをします。
ドライバのバージョンは自分の使っているGPUに合ったものを使用します。

$ sudo add-apt-repository ppa:xorg-edgers/ppa
$ sudo apt-get update
$ apt-cache search 'nvidia-[0-9]+$'
$ sudo apt-get install nvidia-375
$ sudo reboot

dockerのインストール
参考：
docs.docker.com

$ sudo apt-get install \
    apt-transport-https \
    ca-certificates \
    curl \
    software-properties-common
$ curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
$ sudo apt-key fingerprint 0EBFCD88
$ sudo add-apt-repository \
   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) \
   stable"
$ sudo add-apt-repository \
   "deb [arch=armhf] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) \
   stable"
$ sudo apt-get -y update
$ sudo apt-get -y install docker-ce
$ sudo usermod -aG docker $USER

Nvidia-Dockerのインストール

$ sudo apt-get install nvidia-modprobe
$ wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
$ sudo dpkg -i /tmp/nvidia-docker_1.0.1-1_amd64.deb
$ docker volume create -d nvidia-docker --name nvidia_driver_375.66
$ docker volume ls

以上になります。

$ nvidia-docker run -it nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04

f:id:AtsuyaKoike:20181101133959j:plain
あとはこのコマンドが実行できればインストールは終了です。
nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04の部分で、CUDAのバージョン等は自分で指定してください。
https://hub.docker.com/r/nvidia/cuda/