一、下载源码
git clone git://github.com/BVLC/caffe.git
cd caffe/
cp Makefile.config.example Makefile.config
二、编辑Makefile.config
1. 打开
USE_CUDNN := 1
OPENCV_VERSION := 3
WITH_PYTHON_LAYER := 1
2. 删除CUDA_ARCH的两个值
:-gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \
3. 增加HDF5文件引用路径,如下图红体字所示(不一定要添加)
INCLUDE_DIRS :=$(PYTHON_INCLUDE)/usr/local/include /usr/include/hdf5/serial
LIBABRY_DIRS :=$(PYTHON_LIB) /usr/local/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
三、编译caffe框架代码
make all -j8
会有一定数量的报错
1. ./include/caffe/util/math_functions.hpp:7:26: fatal error: glog/logging.h: No such file or directory
问题:缺少glog库
解决:
tar zxvf glog-0.3.3.tar.gz
cd glog-0.3.3/
./configure
make
sudo make install