1. 程式人生 > >Ubuntu16.04上安裝cudnn教程和opencv

Ubuntu16.04上安裝cudnn教程和opencv

https://blog.csdn.net/wang15061955806/article/details/80791112

Ubuntu16.04上安裝cudnn教程

2018年06月24日 14:34:24 閱讀數:3551  

查看了caffe最新版程式的install requirement : https://github.com/BVLC/caffe,這裡我安裝的是CUDA8.0+cudnn5.0(注:其實使用cudnn6.0也可以,但是後來安裝tensorfow-gpu1.0.1時,出現cudnn版本問題,其只支援cudnn5.0,似乎要到tensorflow1.3.0後才支援cudnn6.0,反正現在cudnn5.0也夠用了!)

下載完成後解壓,得到一個 cudn 資料夾,該資料夾下include 和 lib64 兩個資料夾,命令列進入 cudn/include 路徑下,然後進行以下操作:

sudo cp cudnn.h /usr/local/cuda/include/ #複製標頭檔案
  • 1

然後命令列進入 cudn/lib64 路徑下,執行以下命令:

  1. sudo cp lib* /usr/local/cuda/lib64/ #複製動態連結庫
  2. cd /usr/local/cuda/lib64/sudo rm -rf libcudnn.so libcudnn.so.5 #刪除原有動態檔案
  3. sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6 #生成軟銜接
  4. sudo ln -s libcudnn.so.6 libcudnn.so #生成軟連結

安裝完成後可用 nvcc -V 命令驗證是否安裝成功,若出現以下資訊則表示安裝成功:

  1. yhao @yhao-X550VB:~$ nvcc -V
  2. nvcc: NVIDIA (R) Cuda compiler driver
  3. Copyright (c) 2005-2016 NVIDIA Corporation
  4. Built on Tue_Jan_10_13:22:03_CST_2017
  5. Cuda compilation tools, release 8.0, V8.0.61     opencv安裝部分參考部落格: https://blog.csdn.net/cocoaqin/article/details/78163171