參考網站:
http://blog.sina.com.cn/s/blog_a5fdbf010102w7f6.html
http://www.linuxidc.com/Linux/2015-04/116445.htm
ubuntu配置cudnn
-
下載
https://developer.nvidia.com/rdp/cudnn-download
注冊,下載,選擇合適的版本。
用同樣的方法嘗試了cudnn-v3不行,cudnn-v4成功了。
-
解壓
將下載后的文件cudnn-7.0-linux-x64-v4.0-prod.solitairetheme8后綴名改為
cudnn-7.0-linux-x64-v4.0-prod.tgz
-
tar –xvf cudnn-7.0-linux-x64-v4.0-prod.tgz
-
復制
-
$cd cuda/include
-
$sudo cp *.h /usr/local/cuda/include/
-
$cd ../lib64
-
$sudo cp lib* /usr/local/cuda/lib64/
-
-
重新鏈接
-
$cd /usr/local/cuda/lib64/
-
$sudo rm -rf libcudnn.so libcudnn.so.4
-
$sudo chmod u=rwx,g=rx,o=rx libcudnn.so.4.0.7
-
$sudo ln -s libcudnn.so.4.0.7 libcudnn.so.4
-
$sudo ln -s libcudnn.so.4 libcudnn.so
-
$sudo ldconfig
-
-
重新編譯
-
Cd caffe/
-
Make clean
-
Make all –j6
-
成功
-
測試MNIST
-
./example/mnist/train_lenet.sh
-
發現仍然找不到libcudnn.so.4
Lib在編譯時可以放在/usr/local/cuda/lib64
在系統運行時放在/usr/local/lib
因此,按照上述方法,重新將lib* copy到該目錄。
再次運行,成功。48秒。
