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のビルドエラーが解決できない。