CUDA・docker・docker-compose・nvidia-dockerのセットアップメモ

CUDA・docker・docker-compose・nvidia-dockerのセットアップメモ

① Ubuntu 16.04, GeForce GTX TITAN X, CUDA 9.1 @ 2018年4月上旬

CUDA

ここ (https://developer.nvidia.com/cuda-downloads) からファイルをダウンロードして，そこに書かれている手順でインストールします．ここは失敗しない．

$ wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ apt-get update
$ apt-get install cuda

Driver

ここ (http://www.nvidia.com/Download/index.aspx) からダウンロードしたいのですが，TITAN X, Ubuntu 16.04 を選択するとなぜか存在しないと言われます．そこで以下の手順でdriverをインストールします．

$ add-apt-repository ppa:graphics-drivers/ppa
$ apt-get update
$ apt-get install nvidia-390

動作確認: nvidia-smi

docker

$ wget -qO- https://get.docker.com/ | sh

動作確認: docker run --rm hello-world

ただし，nvidia-dockerが入らなかったのでダウングレードしました (18.04.0 -> 18.03.0)．

$ apt-get install docker-ce=18.03.0~ce-0~ubuntu

私が勘違いしていたようで，事前にdockerをインストールする必要はなく，nvidia-dockerするだけで良かったようです．

nvidia-docker

上記のようにダウングレードしておけばapt-getでインストールできます．
事前にdockerをインストールする必要はなく，下記のコマンドでdockerも含めてインストール可能なようです．

https://nvidia.github.io/nvidia-docker/ を参考にRepositoryを追加します．

$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
$ curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update

そしてインストール．

$ apt-get install nvidia-docker2
$ pkill -SIGHUP dockerd

動作確認: nvidia-docker run --rm nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04 nvidia-smi

docker-compose

$ pip install docker-compose

② Ubuntu 18.04, Quadro K2200, CUDA 9.1 @ 2018年5月上旬

CUDA

現時点では18.04をサポートしていません．（すぐにサポートされると思われますが）
そこで，公式からダウンロードした17.04の用のRunfileを用いてインストールしました．

$ apt-get install g++-6
$ apt-get install nvidia-cuda-toolkit

$ wget https://developer.nvidia.com/compute/cuda/9.1/Prod/local_installers/cuda_9.1.85_387.26_linux
$ ./cuda_9.1.85_387.26_linux --override

インストール中に「サポートしていないsystemにインストールするか？」と聞いてくるのでyesと答えます．次に「Driverをインストールするか」と聞かれますが後で入れるのでnoと答えます．
あとは適当に答えるとインストールできます．
(参考: https://askubuntu.com/questions/1028830/install-cuda-on-ubuntu-18-04)

Driver

$ apt-get install nvidia-driver-390

nvidia-docker

Dockerは事前にインストールしなくても，①の手順でnvidia-docker2をインストールできました．