环境:
Cuda: 8.0.61
Cudnn: 6.0.21
前期准备啥的就不赘述了,下面正式开始安装caffe
1 首先在你要安装的路径下 clone :
git clone https://github.com/BVLC/caffe.git
2 进入 caffe ,将 Makefile.config.example 文件复制一份并更名为 Makefile.config ,也可以在 caffe 目录下直接调用以下命令完成复制操作 :
sudo cp Makefile.config.example Makefile.config
3 复制一份的原因是编译 caffe 时需要的是 Makefile.config 文件,而Makefile.config.example 只是caffe 给出的配置文件例子,不能用来编译 caffe。
sudo gedit Makefile.config
4 修改 Makefile.config 文件内容:
a 应用 cudnn
将
#USE_CUDNN := 1
修改成:
USE_CUDNN := 1
b 应用 opencv 版本
将
#OPENCV_VERSION := 3
修改为:
OPENCV_VERSION := 3
c 使用 python 接口
将
#WITH_PYTHON_LAYER := 1
修改为
WITH_PYTHON_LAYER := 1
d 修改 python 路径
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
5 然后修改 caffe 目录下的 Makefile 文件:
将:
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替换为:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
参考其他博客有下面这一步,但是本人修改这个后报错了,本人没有修改这个。
将:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
改为:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
6 然后修改 /usr/local/cuda/include/host_config.h 文件 :
将
#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
改为
//#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
7 OK ,可以开始编译了,在 caffe 目录下执行 :
make all -j8
8 安装 pycaffe notebook 接口环境
a 编译 pycaffe :
cd caffe
sudo make pycaffe -j8
9 最重要的是:要添加环境变量哦
vim ~/.bashrc # 最后一行添加
export PYTHONPATH=/home/xxxx/Documents/caffe/python:$PYTHONPATH
然后再重新加载:source ~/.bashrc