UbuntuにcuDNNをインストール

メモとして残しておきます。

cuDNNとは

NVIDIAが公開しているDeep Learning用のライブラリである。このライブラリを使うとCaffeやChainerなどのDeep Learning用のソフトウェアの速度が向上する。

cuDNNのインストール

以下のリンクからダウンロード

*アクセスするにはメンバー登録が必要である

cuDNN https://developer.nvidia.com/rdp/cudnn-download

具体的にはdebパッケージ３つをダウンロード
・runtime library
・developer library
・code samples and user guide

CUDA https://developer.nvidia.com/cuda-80-ga2-download-archive

~$ sudo dpkg -i libcudnn7_7.4.1.5-1+cuda9.0_amd64.deb
~$ sudo dpkg -i libcudnn7-dev_7.4.1.5-1+cuda9.0_amd64.deb
~$ sudo dpkg -i libcudnn7-doc_7.4.1.5-1+cuda9.0_amd64.deb

cuDNNのライブラリをCUDAがインストールされているディレクトリにコピーする。

~$ tar xvf cudnn-9.0-linux-x64-v7.4.15.tgr
~$ sudo cp -a cuda/include/cudnn.h /usr/local/cuda/include/
~$ sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64/
~$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
~$ cp -r /usr/src/cudnn_samples_v7/ $HOME
~$ cd $HOME/cudnn_samples_v7/mnistCUDNN
~$ make clean && make

これだとエラーが出たので以下のコマンドを実行

~$ sudo apt install gcc-6
~$ sudo apt install g++-6

~$ sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
~$ sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
~$ cd $HOME/cudnn_samples_v7/mnistCUDNN
~$ make clean && make
./mnistCUDNN

+αとしてcudaのversionアップの方法である

~$ sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub
~$ sudo apt-get update
~$ sudo apt-get install cuda

*/usr/local/にインストールしたcudaがあります。

~/.bashrc の内容を現在のversionに書き換え

~$ sudo vim ~/.bashrc
~$ source ~/.bashrc