nvidia docker setup - eiichiromomma/CVMLAB GitHub Wiki

(nvidia-docker) Setup

nvidia-dockerが使えるようになるまでの作業メモ

環境

  • Ubuntu 16.04 LTS 64bit
  • nVidia Geforce 1080

Ubuntuの環境構築手順(CUDA)

DeepLearning CUDAを参照。

Dockerのインストール

sudo apt-get install apt-transport-https ca-certificates
sudo apt-key adv --keyserver hkp://ha.pool.sks-keyservers.net:80 --recv-keys 58118E89F3A912897C070ADBF76221572C52609D
sudo vim /etc/apt/sources.list.d/docker.list

中身は

deb https://apt.dockerproject.org/repo ubuntu-xenial main

にする。あとはインストール

sudo aptitude update
sudo aptitude install docker-engine
sudo service docker start

nvidia-dockerのインストール

wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0-rc.3/nvidia-docker_1.0.0.rc.3-1_amd64.deb
sudo dpkg -i /tmp/nvidia-docker_1.0.0.rc.3-1_amd64.deb 
sudo nvidia-docker run --rm nvidia/cuda nvidia-smi

でnvidia-smiの結果が出てくれば成功。

使い方

基本的にdockerと同じ。Ubuntuではsudoが必要。