TensorFlow > cuDNN v4.0のアンインストール / cuDNN v5.1のインストール

動作環境

Ubuntu 14.04 LTS desktop amd64
GeForce GTX 750 Ti
ASRock Z170M Pro4S [Intel Z170chipset]

http://qiita.com/7of9/items/94c33411891decfdd8d4
にて出たエラーの対応。

経緯としては

• TensorFlow v0.8とCUDA v7.5, cuDNN v4をインストール
• TensorBoardが表示されない問題対応のため、TensorFlow v0.11をインストール
• 上記リンクのエラーが出ることに気づいた

cuDNNのバージョンを上げてみる。

cuDNN v4 のアンインストール

uninstallの方法は検索してもはっきりしなかった。
インストールと反対の方法をとってみた。

$sudo rm /usr/local/cuda/include/cudnn.h

$ls /usr/local/cuda/lib64/libcudnn*
/usr/local/cuda/lib64/libcudnn.so    /usr/local/cuda/lib64/libcudnn.so.4.0.7
/usr/local/cuda/lib64/libcudnn.so.4  /usr/local/cuda/lib64/libcudnn_static.a

$ sudo rm /usr/local/cuda/lib64/libcudnn*

cuDNN v5.1のインストール

$ tar xvzf cudnn-7.5-linux-x64-v5.1.tgz 
cuda/include/cudnn.h
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.5
cuda/lib64/libcudnn.so.5.1.3
cuda/lib64/libcudnn_static.a
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
$ ls -l /usr/local/cuda/lib64/libcudnn*
-rwxr-xr-x 1 root root 60696704 10月  9 11:18 /usr/local/cuda/lib64/libcudnn.so
-rwxr-xr-x 1 root root 60696704 10月  9 11:18 /usr/local/cuda/lib64/libcudnn.so.5
-rwxr-xr-x 1 root root 60696704 10月  9 11:18 /usr/local/cuda/lib64/libcudnn.so.5.1.3
-rw-r--r-- 1 root root 59715990 10月  9 11:18 /usr/local/cuda/lib64/libcudnn_static.a