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