[TOC]

简介

caffe是一个清晰，可读性高，快速的深度学习框架。作者是贾扬清，加州大学伯克利的ph.D，现就职于Facebook。caffe的官网是http://caffe.berkeleyvision.org/。

安装

环境

• Ubuntu 16.04 LTS
• CUDA8.0

参考资料

• 官方安装指南：Ubuntu Installation

注： 对于 Ubuntu >= 17.04 的Ubuntu系统，安装将变得极为简单，通过Ubuntu的包管理器安装即可，这里不作说明。

准备

基础环境配置

• 系统安装，可移步本人博客：U盘系统盘制作与系统安装（详细图解）
• CUDA安装，可移步本人博客：Ubuntu 16.04 LTS + CUDA8.0 + cudnn6.0 或 NVIDIA DIGITS 学习笔记（NVIDIA DIGITS-2.0 + Ubuntu 14.04 + CUDA 7.0 + cuDNN 7.0 + Caffe 0.13.0）

依赖库安装

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libatlas-base-dev

构建与安装

获取源码

从 这里 （或终端输入命令 git clone https://github.com/BVLC/caffe.git ）下载caffe源码并解压，修改文件夹名为 caffe 。

For Python

如果你需要使用Caffe 的 Python 接口，你还需要安装Python的一些库，进入caffe根目录下的python目录，执行：

for req in $(cat requirements.txt); do sudo pip install $req; done

然后添加 PYTHONPATH 环境变量：终端输入 sudo gedit ~/.bashrc 打开 .bashrc 文件，文件末尾加入（注意修改为你的caffe绝对路径）：

#caffe python
export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

如果你使用 Anaconda ，下面修改 Makefile.config 文件时，注意自行 启用 Anaconda，约73行。

修改配置文件

这一步修改 Makefile.config 文件，以满足自己的需求，如：

#启用cuDNN加速
USE_CUDNN := 1

另外，在Ubuntu16.04上安装caffe还需要修改以下几处，否则会报出各种错误，具体参见问题解决：

将如下几处代码：

#第一处：CUDA PATH
CUDA_DIR := /usr/local/cuda

#第二处：PYTHON PATH
# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include

改为：

#第一处：CUDA PATH，改为：
CUDA_DIR := /usr/local/cuda-8.0

#第二处：PYTHON PATH，添加hdf5库，改为：
# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/

构建安装

进入caffe的根目录，终端执行如下命令，不报错误，代表安装成功！( -j$(nproc) 代表使用最大的线程编译，当然也可以手动指定，如 make -j4 ； make py 是用来编译caffe的python接口，如不需要可忽略。)

make all -j$(nproc)
make pycaffe -j$(nproc)
make test -j$(nproc)
make runtest -j$(nproc)

添加环境变量

终端输入 sudo gedit ~/.bashrc 打开 ".bashrc" 文件，在文件末尾加入如下代码并保存：

# Set Caffe environment
export CAFFE_ROOT=/home/liu/sfw/dlapp/caffe/

#caffe python
export PYTHONPATH=/home/liu/sfw/dlapp/caffe/python/:$PYTHONPATH

然后重新打开一个终端，或者输入 source ~/.bashrc 加载新的环境变量。

验证Python接口

为验证安装成功，终端输入 ipython 或 python 进入Python解释器环境，输入： import caffe ，不报错误代表caffe的Python接口可以使用。

问题解决

安装

hdf5

1. 现象

In file included from src/caffe/solvers/sgd_solver.cpp:5:0:
./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: No such file or directory
compilation terminated.
Makefile:581: recipe for target '.build_release/src/caffe/solvers/sgd_solver.o' failed

2. 解决

首先确保前面安装依赖库时安装了 libhdf5-serial-dev ，然后修改 Makefile.config 文件（约95行）：

# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/

No module named caffe

请确认是否执行 make pycaffe 且没有报错，如果是请再检查环境变量的设置。