caffe 配置 札记

cudnn的配置

1. 安装前请去先官网下载cuDNN (cudnn-7.0-linux-x64-v3),建议安装v3，v4有些问题。

将cudnn-7.0-linux-x64-v3解压后会有两个文件夹include和lib64，复制其中文件到系统库

sudo cp include/cudnn.h /usr/local/include
sudo cp lib64/libcudnn.* /usr/local/lib

2. 链接cuDNN的库文件

sudo ln -sf /usr/local/lib/libcudnn.so.7.0.64 /usr/local/lib/libcudnn.so.7.0
sudo ln -sf /usr/local/lib/libcudnn.so.7.0 /usr/local/lib/libcudnn.so
sudo ldconfig -v

3. 在Makefile.config中启用CUDNN，去掉注释

USE_CUDNN := 1

参考资料

参考资料2