假设所有依赖(见 http://blog.csdn.net/junglezax/article/details/45741137 《caffe install at centos 依赖软件包安装》),环境变量,库路径,软链接都已装完做好
假设安装在 ~/caffe-master
假设 python 使用 anaconda, 安装在 /root/green/anaconda
cd ~/ca
ffe-master
cp Makefile.config.example Makefile.config
vi Makefile.config
USE_CUDNN := 1
CUDA_DIR := /usr/local/cuda
BLAS := atlas
BLAS_LIB := /usr/lib64/atlas/
PYTHON_INCLUDE := /root/green/anaconda/include \
/root/green/anaconda/include/python2.7 \
/root/green/anaconda/lib/python2.7/site-packages/numpy/core/include
PYTHON_LIB := /root/green/anaconda/lib
如果不用 GPU,则
CPU_ONLY := 1
#USE_CUDNN := 1
如果需要 matcaffe 接口
MATLAB_DIR := /usr/local/MATLAB/R2013b/
源代码修正(我用的代码有点旧,如果是 2015年5月以后的可能不需要)
vi src/caffe/test/test_power_layer.cpp
line 77:
GradientChecker<Dtype> checker(1e-3, 1e-2, 1701, 0., 0.01);
把 (1e-2 改成 (1e-3
make clean
make all
make test
make runtest
编译 matcaffe 接口
make matcaffe