ubuntu配置cudnn


參考網站:

http://blog.sina.com.cn/s/blog_a5fdbf010102w7f6.html

http://www.linuxidc.com/Linux/2015-04/116445.htm

ubuntu配置cudnn

  1. 下載

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

注冊,下載,選擇合適的版本。

用同樣的方法嘗試了cudnn-v3不行,cudnn-v4成功了。

  1. 解壓

將下載后的文件cudnn-7.0-linux-x64-v4.0-prod.solitairetheme8后綴名改為

cudnn-7.0-linux-x64-v4.0-prod.tgz

  1. tar –xvf cudnn-7.0-linux-x64-v4.0-prod.tgz
  1. 復制
    1. $cd cuda/include
    2. $sudo cp *.h /usr/local/cuda/include/
    3. $cd ../lib64
    4. $sudo cp lib* /usr/local/cuda/lib64/
  2. 重新鏈接
    1. $cd /usr/local/cuda/lib64/
    2. $sudo rm -rf libcudnn.so libcudnn.so.4
    3. $sudo chmod u=rwx,g=rx,o=rx libcudnn.so.4.0.7
    4. $sudo ln -s libcudnn.so.4.0.7 libcudnn.so.4
    5. $sudo ln -s libcudnn.so.4 libcudnn.so
    6. $sudo ldconfig
  3. 重新編譯
    1. Cd caffe/
    2. Make clean
    3. Make all –j6

成功

  1. 測試MNIST
    1. ./example/mnist/train_lenet.sh

發現仍然找不到libcudnn.so.4

Lib在編譯時可以放在/usr/local/cuda/lib64

在系統運行時放在/usr/local/lib

因此,按照上述方法,重新將lib* copy到該目錄。

再次運行,成功。48秒。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM