nvidia-dockerインストール、コンテナ等メモ
基本的に公式サイトの説明とかの通りにやるだけなんだけど自分用にメモ。
ホストOS:ubuntu 16.04LTS
dockerのインストール
参考: Get Docker for Ubuntu - Docker
# https通信で公開鍵ダウンロードするため apt-get update apt-get install curl ca-certificates # dockerレポジトリの公開鍵の登録 curl -fsSL https://yum.dockerproject.org/gpg | apt-key add - # add-apt-repositoryコマンドのインストール apt-get install software-properties-common # dockerレポジトリ追加 add-apt-repository "deb https://apt.dockerproject.org/repo/ ubuntu-$(lsb_release -cs) main" # aptでhttps通信するため apt-get install apt-transport-https # dockerのインストール apt-get update apt-get install docker-engine
NVIDIAドライバのインストール
これが一番ハマって大変な場合が多い気がするが、ここでは省略。どうにかうまいことインストールする。
NVIDIA Dockerのインストール
# debパッケージのダウンロード(2017/1/22現在で最新のもの) curl -LO https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0/nvidia-docker_1.0.0-1_amd64.deb # インストール dpkg -i nvidia-docker_1.0.0-1_amd64.deb rm nvidia-docker_1.0.0-1_amd64.deb # 動作確認(GPU情報が表示されると良い) nvidia-docker run --rm nvidia/cuda nvidia-smi
バージョンについては下記ページでチェックして適宜ダウンロードするものを確認する。
Releases · NVIDIA/nvidia-docker · GitHub
NVIDIA Dockerコンテナメモ
便利なものがあれば随時追記予定。
- https://hub.docker.com/u/nvidia/
nvidia公式コンテナ。CUDAやCuDNNのバージョン、OSの種類を選べる。 - https://hub.docker.com/u/kaixhin/
イギリスの研究者作成のコンテナ。色々あって便利。