Docker

ubuntu16でNVIDIA Docker2を使ってみる(CUDA,cDNN)

2018/04/23


ubuntuにNVIDIAのドライバーをインストール


Download→Linux→x84_64→Ubuntu→16.04→dev(network)

apt-keyコマンド実行時に下記のエラーが出力された場合
gpgkeys: protocol `https' not supported
sudo apt-get install gnupg-curl
gnupgをインストールすれば解決します。

再起動後、nvidia-smiコマンドが使えるようになっている。
cat /proc/driver/nvidia/version

gpuの使用状況の確認

Dockerをインストール

一般ユーザにもdockerコマンドの実行権限を付与する

コマンドを一度抜けて(exit)、再度一般ユーザでログインするとdockerコマンドが使えます。

NVIDIA docker2のインストール

下記のコマンドを実施してnvidia-smi -lの結果が出力されれば成功。

nvidia-docker-composeのインストール

nvidia-docker-composeコマンドは使えなくなっているみたいです。

docker-composeのデフォルトのruntimeをdaemon.jsonファイルで設定し、dockerを再起動します。

docker-compose upを実行してnvidia-smiの結果が出力されれば成功。
version2.3からruntimeオプションが使えるとDocumentにありましたが、できなかったので。。。

cuDNNのインストール


cuDNN 7.1 for CUDA 9.1のdebパッケージをダウンロードする(アクセスするにはメンバー登録が必要)。
以下の3個のdebパッケージをダウンロードする。

cuDNN v7.1.* Runtime Library for Ubuntu16.04 (Deb)
cuDNN v7.1.* Developer Library for Ubuntu16.04 (Deb)
cuDNN v7.1.* Code Samples and User Guide for Ubuntu16.04 (Deb)

cuDNNのインストール

以上です

-Docker
-, , ,