(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
完成。