关于CUDA8.0和cuDNN6.0、opencv3.1的配置可以查看我前面的博客:
本文参考博客:https://blog.csdn.net/yhaolpz/article/details/71375762
1、下载、解压Caffe
建议直接去https://github.com/BVLC/caffe下载zip压缩包,因为使用git命令clone的话比较慢。
解压Caffe至想安装的目录。
2、修改Makefile.config文件
进入解压好的Caffe文件夹,将Makefile.config.example复制一份并改名为Makefile.config:
sudo cp Makefile.config.example Makefile.config
修改Makefile.config文件:
sudo gedit Makefile.config
修改如下:
1)应用cuDNN:
将
#USE_CUDNN := 1
修改成:
USE_CUDNN := 1
2)应用opencv:
将
#OPENCV_VERSION := 3
修改为:
OPENCV_VERSION := 3
3)使用Python接口:
将
#WITH_PYTHON_LAYER := 1
修改为
WITH_PYTHON_LAYER := 1
4)修改路径:
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
3、修改Makefile文件
进入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
4、修改CUDA的host_config文件
修改/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!
5、编译
make all -j8
sudo make runtest -j8