Ubuntu16.04深度学习环境配置——Caffe的安装(CUDA8.0+cuDNN6.0)

关于CUDA8.0和cuDNN6.0、opencv3.1的配置可以查看我前面的博客：

• Ubuntu16.04深度学习环境配置——NVIDIA显卡驱动、CUDA8.0、cuDNN v6.0
• Ubuntu16.04配置opencv3.1(已安装CUDA8.0)

本文参考博客：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