caffe ubuntu 16.04 cpu版安装方法

安装依赖项

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 libatlas-base-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libblas-dev

安装opencv,详细方法可见本人文章opencv linux安装与编译.

python安装

caffe支持python2.7版本,其他版本是否支持不太清楚,网络上给的安装方法基本都是基于python2.7版本上的.

python --version

查看系统默认python版本.

安装python-dev依赖包

sudo apt-get install python-dev

安装caffe

下载caffe

sudo apt-get install git
git clone git://github.com/BVLC/caffe.git

编译caffe

cd caffe/
cp Makefile.config.example Makefile.config
sudo vim Makefile.config

对Makefile.config文件做如下修改:

• 因为CPU MODE, 所以在CPU_ONLY := 1前面的#要去掉.

• 在opencv 3.x版本下编译,所以在OPENCV_VERSION := 3前面的#要去掉.

• 两个路径要改成这样:(添加后面的两个hdf5的路径, 否则编译时报hdf5错误)

  # Whatever else you find you need goes here.
  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/hdf5/serial
  

下面开始编译:

make all
make test
make runtest

这个make默认是用CPU单核运算,如果想要快一点, 比如我想使用四核, 在make后面加上-j4标签.
如果上面4行某一行报错之后想要重试,建议先make clean再重新开始.

编译python接口

通常为了更好的调用caffe，我们往往需要写代码，caffe具有Python和C++接口，我们较为常用Python接口，这里我将展示如何编译Python接口。

安装pip

sudo apt-get install python-pip

安装Python接口依赖库.在caffe根目录下，有个python文件夹，文件夹里面有个requirements.txt，里面有需要的依赖库和版本信息，按照其安装即可，在安装前，需要先安装fortran编辑器(gfrotran)，因为安装scipy库时需要它，命令如下：

sudo apt-get install gfortran
cd ~/caffe/python
for req in $(cat requirements.txt); do pip install $req; done

安装结束后，可以执行如下语句验证：

sudo pip install -r requirements.txt

可以看到，安装成功的，都会显示Requirement already satisfied，没安装成功的，会继续进行安装。

将caffe根目录下的python文件夹加入到环境变量

先打开配置文件bashrc

sudo vim ~/.bashrc

在文件的最后面添加

export PYTHONPATH=/home/robert/Documents/caffe/python:$PYTHONPATH

然后执行更新配置(有时更新不起作用时，可以重启电脑就会有效)：

sudo ldconfig

编译python接口

cd ../../caffe/
make pycaffe

验证python接口

python
import caffe

如果import成功,则安装成功.