caffe环境搭建与初体验(一)

caffe环境搭建

1. 下载caffe

caffe源代码下载地址： https://github.com/BVLC/caffe

将下载的caffe源码压缩包解压并移动到home目录下：

cd ~/Downloads
unzip caffe-master.zip
sudo mv ./caffe-master ~/caffe-master

2. 修改Makefile.config配置文件

复制一份Makefile.config文件并使用gedit打开修改配置：

cd ~/caffe-master
sudo cp Makefile.config.example Makefile.config
sudo gedit Makefile.config

主要修改如下节点：

• #USE_CUDNN:=1 改为 USE_CUDNN：=0 (无显卡则不使用CUDN，注意去掉前边的#符号)
• #CPU_ONLY := 1 改为 CPU_ONLY := １(去除注释，仅使用cpu)
• #OPENCV_VERSION := 3 (看你是否使用opencv)
• #WITH_PYTHON_LAYER :=1 改为 WITH_PYTHON_LAYER :=1（python接口）

ubuntu16.04版本以下文件包含位置有变化，所以以下两处一定要做更改：

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include 改为：

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 改为：

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-master目录下的makefile文件，做如下修改：

NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS) 修改为：

NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

4. 编译caffe

在caffe-master目录运行终端，输入：

sudo make all -j4

5. 测试caffe是否安装成功

运行以下指令：

sudo make runtest

6. 常见问题汇总

６.１ /usr/bin/ld: 找不到 -lcudnn

• 设置USE_CUDNN := １，但是却没有安装cudnn禁用USE_CUDNN或者安装cudnn(注意：ubuntu上安装这个可能会遇到各种问题，建议直接注释掉不用)

６.2 No module named lmdb

• sudo apt-get install liblmdb-dev

(注意：有可能会发生一些使用apt-get难以解决的依赖问题，可以直接sudo apt-get install aptitude,安装aptitude，aptitude在解决依赖方面比较友好)

６.３ No module named _caffe

• 安装了caffe之后，发生这种情况，一般是环境变量没设置

# 设置环境变量
$ sudo  gedit /etc/profile
# 添加环境变量：
    export PYTHONPATH=${HOME}/caffe-master/distribute/python:$PYTHONPATH
    export LD_LIBRARY_PATH=${HOME}/caffe-master/build/lib:$LD_LIBRARY_PATH
$ source /etc/profile # 使环境变量生效
$ echo $<环境变量名> # 查看环境变量

６.4 fatal error: hdf5.h: 没有那个文件或目录

- Step 1： 在Makefile.config文件中，添加/usr/include/hdf5/serial/ 到 INCLUDE_DIRS，也就是把下面第一行代码改为第二行代码：
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
- Step 2： 在Makefile文件的第173行，把 hdf5_hl 和hdf5修改为hdf5_serial_hl 和 hdf5_serial，也就是把下面第一行代码改为第二行代码：   
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