(1)下載
安裝完cuda后,要想利用cudnn的優化能力,需要安裝深度學習包cudnn,
首先下載最新的cudnn,https://developer.nvidia.com/cudnn,我的是v4 ,適合cuda7.0及以上(我的是7.5)。
(2)解壓
得到兩個文件夾,一個包含頭文件cudnn.h,另一個包含cudnn的動態鏈接庫(.so)和靜態鏈接庫(.a)。
(3)配置
將cudnn.h復制到usr/local/include中,將三個動態鏈接庫拷貝到usr/local/lib中。紅框中是cudnn的動態庫,綠框中是opencv的imgcodecs的動態庫。
然后軟鏈接,如下命令:
$ sudo ln -sf /usr/local/lib/libcudnn.so.4.0.07/usr/local/lib/libcudnn.so.4 $ sudo ln -sf /usr/local/lib/libcudnn.so.4/usr/local/lib/libcudnn.so $ sudo ldconfig -v
完成。