下述教程均基于Ubuntu14.04,且默认已安装好caffe所需要的所有依赖库以及cuda。如未安装,请先找教程安装好。
1. 首先从Github上下载caffe。
2. 解压缩,并创建Makefile.config文件。
unzip caffe-master.zip
cd caffe-master
cp Makefile.config.example Makefile.config
3.编辑Makefile.config,取消注释 USE_CUDNN := 1
4. https://developer.nvidia.com/cudnn 注册账号(可能需要一晚上的验证)下载最新(v2以上)版的cuDNN。
5. 安装cuDNN(以R1版本为例,注意换为自己下载的包名)
tar -xzvf cudnn-6.5-linux-R1.tgz
cd cudnn-6.5-linux-R1 sudo cp lib* /usr/local/cuda/lib64/ sudo cp cudnn.h /usr/local/cuda/include/
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6.5
sudo chmod u=rwx,g=rx,o=rx libcudnn.so.6.5.18 sudo ln -s libcudnn.so.6.5.18 libcudnn.so.6.5 sudo ln -s libcudnn.so.6.5 libcudnn.so6.build
make all
make test
make runtest