読者です 読者をやめる 読者になる 読者になる

Deep LearningのGPU環境構築

(2016/3/4追記)この記事の内容は古くなっています。
tensorflow 0.7.1, chainer 1.7.0の場合、CUDA 7.5 + cuDNN v4を設定すること。

Ubuntu 14.04 LTS 64bitのインストール

システムの更新

$ sudo apt-get update
$ sudo apt-get dist-upgrade

リセット

nVidiaドライバのインストール

システム設定→ソフトウェアとアップデート→追加のドライバー
NVIDIA binary driver を選択
変更の適用

CUDA toolkitのインストール
事前にダウンロードする

$ sudo dpkg -i cuda-repo-ubuntu1404_7.0-local_7.0-28_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

.bashrcに追記

export CUDA_HOME=/usr/local/cuda-7.0
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

cuDNNライブラリ

$ wget http://developer.download.nvidia.com/compute/redist/cudnn/v4/cudnn-7.0-linux-x64-v4.0-rc.tgz
$ tar -xzvf ./cudnn-7.0-linux-x64-v4.0-rc.tgz
$ sudo mkdir /usr/local/cudnn-v4-rc
$ sudo cp -r cuda /usr/local/cudnn-v4-rc

.bashrcに追記

export CUDNN_HOME=/usr/local/cudnn-v4-rc/cuda
export CPATH=$CUDNN_HOME/include:$CPATH
export LIBRARY_PATH=$CUDNN_HOME/lib64:$LIBRARY_PATH
export LD_LIBRARY_PATH=$CUDNN_HOME/lib64:$LD_LIBRARY_PATH

ファイルのコピー

$ cp /usr/local/cudnn-v4-rc/cuda/include/cudnn.h $CUDA_PATH/include
$ cp /usr/local/cudnn-v4-rc/cuda/lib64/libcudnn* $CUDA_PATH/lib64

※追記
tensorflow 0.7.0のバイナリだと実行時にCUDNNのエラーがでる。
v3-prodのcuDnnに変えれば実行できる。
v4ができそうな記載あるからソースからビルドするも、
bazelのビルドエラーが解決できない。