NVIDIA DIGITS 6をdockerで実行

システム構成

• Ubuntu Server 16.04.3 LTS
• NVIDIA GeForce GTX 1080

インストール

Docker CE

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

nvidia-docker

$ 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*.deb && rm /tmp/nvidia-docker*.deb
$ sudo nvidia-docker run --rm nvidia/cuda nvidia-smi

起動

$ sudo nvidia-docker run  -d --restart=always -d  -p 5000:5000  -p 6006:6006  -v /{ホスト側のパス}/digits/jobs:/jobs  -v /{ホスト側のパス}/digits/data:/data  nvidia/digits:6.0

参照

• nvidia-docker
• NVIDIA DIGITS