Ubuntu16.04にnvidia-dockerをインストールする方法

はじめに

dockerのversionも大分上がったので再度インストール設定をまとめました。
AWS EC2 P2インスタンスで確認

nvidiaドライバインストール

アップデート

sudo apt update -y
sudo apt upgrade -y

unable to resolve hostと言われた場合実行(今回は言われた)

sudo sh -c 'echo 127.0.1.1 $(hostname) >> /etc/hosts'

Repositoryを追加して、ドライバインストール。
ドライバのVersionはそれぞれの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インストール

基本的に公式サイトに従う。

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
# ログインし直すとsudoしなくてもdockerコマンドが使える

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 nvidia-smi

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 375.66                 Driver Version: 375.66                    |
|-------------------------------+----------------------+----------------------+
以下略