udnn安裝注意點:
cudnn的安裝其實很簡單,關鍵點是一定要安裝cuda對應的cudnn包,本機中安裝的cuda7.5所以對應的cudnn為v5.1這很重要,我就是安裝錯了版本,導致后面caffe的編譯總是出錯。
cudnn安裝步驟:
1、從官網上下載cudnn的安裝包。
2、將安裝包解壓,將此安裝包放在home路徑下即可,並在當前路徑下進行解壓,解壓后的文件夾名為cuda。
3、在終端上編輯如下代碼:
$ cd cuda/include $ sudo cp *.h /usr/local/cuda/include/ (注意這里是你自己的cuda的安裝地址,復制*.h文件)
4、關閉終端並重新打開(也可以返回最開始的目錄)
$ cd cuda/lib64 $ sudo cp libcudnn* /usr/local/cuda/lib64/
5、終端運行:
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
運行faster rcnn代碼提示:
Loaded runtime CuDNN library: 7.1.3 but source was compiled with: 7.4.1. CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library. If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration.
遂將cudnn卸載:
sudo rm -rf /usr/local/cuda/include/cudnn.h sudo rm -rf /usr/local/cuda/lib64/libcudnn*
將cudnn卸載后,重新安裝cudnn
轉自:https://blog.csdn.net/u010480194/article/details/54287727