caffe python接口_Ubuntu16.04 安装 caffe python 接口

前言

这篇笔记记录自己在 Ubuntu16.04 上面安装CPU 模式 caffe python 接口流程，备份下来，以便学习。主要参考如下两篇博客，谢谢原作者的分享 👍：

1. 安装所需依赖

caffe python 接口依赖 numpy 组件，numpy 的安装可以用 pip 安装，没有安装 pip 的话可以用下面的命令先安装 pip，若提示权限不足，可以前加 sudo，输入管理员密码安装

apt-get install pip

然后用下面命令安装 numpy：

pip install numpy

继续安装其他依赖(这些安装的依赖是照着开头两位博主的教程，未仔细研究每个包的具体作用，后面希望可以研究清楚)：

apt-get install libboost-all-dev

apt-get install libprotobuf-dev

apt-get install libleveldb-dev

apt-get install libsnappy-dev

apt-get install libopencv-dev

apt-get install libhdf5-serial-dev

apt-get install protobuf-compiler

apt-get install libgflags-dev

apt-get install libgoogle-glog-dev

apt-get install liblmdb-dev

apt-get install libatlas-base-dev

2. 安装 OpenCV3

我是参考的这篇教程，谢谢原博主的分享 👍

注：根据这篇教程 https://blog.csdn.net/u010193446/article/details/53259294，不安装 OpenCV 也不影响 caffe 的安装(若未安装OpenCV，则第三节对 OpenCV 相关配置更改可以不做)，因为我之前已安装 OpenCV3，所以此处无法验证，如果有验证过的同学，麻烦留言告知，谢谢👊！

3. 安装 caffe CPU 版本

先下载 caffe 源码，使用 git 下载，未安装 git 的话，可以使用 apt-get install git 安装。假设 caffe 下载目录为 ~

cd ~

git clone https://github.com/BVLC/caffe

cd caffecp Makefile.config.example Makefile.config

caffe 原目录下面有个 Makefile.config.example 文件作为编译安装配置的示例，可以基于此文件进行针对性修改，以便在自己的机器上正确安装，下面介绍几点关键的修改地方：

1. 取消对 CPU_ONLY:=1 的注释，如下图(左为修改前，右为修改后)：

2. 取消对 OPENCV_VERSION:=3 的注释，如下图：

3. 对 PYTHON_INCLUDE 的注释和修改

这部分涉及到编译 caffe python 接口时的头文件和库文件。如果安装的是 python2，则修改默认的 PYTHON_INCLUDE，使之为正确的python目录地址：

如果安装的是 python3，则需将 python2 对应的 PYTHON_INCLUDE注释(也就是上图那部分👆)，并取消对 python3 的 PYTHON_INCLUDE注释，同时修改内容。需要注意的时，PYTHON_INCLUDE 的默认头文件位置以及 PYTHON_LIBRARIES 的默认库位置，可能和你的安装位置不一样，需要进行修改；

例如我的 numpy 目录和默认的就不同，同时 libboost_python 版本也和默认的不同，均需要修改。若不清楚 numpy 的目录哪里，可以使用 locate 命令查询得到：

locate -e arrayobject.h

注意：上面 PYTHON_LIBRARIES 配置未修改前是 boost_python3， 对编译安装 caffe 无影响，但在安装 python 接口时会提示找不到对应的库，对此我的修改方式是其改为 boost_python35，也可以在 libboost_python35.so 所在目录里创建一个符号链接，命令如下：

ln -s libboost_python35.so libboost_python3.so

boost 的默认安装目录为 /usr/lib 或 /usr/local/lib，根据自己机器的实际情况查询即可；

locate -e libboost_python

4. 配置引用文件路径

改为

即在 INCLUDE_DIRS 后加上 /usr/include/hdf5/serial，在 LIBRARY_DIRS 后加上 /usr/lib/x86_64-linux-gnu/hdf5/serial

5. 执行编译安装

makeallmaketestmake runtest

上述每个步骤执行完会依次显示如下内容(具体内容可能有所不同，没有错误显示，即为安装成功，安装时未能保存截图，此处引用 https://blog.csdn.net/g11d111/article/details/78141202 中的截图作为示例，谢谢原作者的分享)。

注意：如果执行失败，则需要执行 make clean 清理安装，解决问题后再执行安装；

4. 编译 python 接口

1. 安装 python 接口依赖库

进入到 caffe 目录下的 python 目录，内有 requirement.txt 文件，列出了依赖的 python 库和版本信息，安装即可；

apt-get installgfortran

cd~/caffe/pythonfor req in $(cat requirements.txt); do pip install $req; done

2. 将上面的 python 目录加入到环境变量

编辑 ~/.bashrc 文件

vi ~/.basrc

文件末尾添加

export PYTHONPATH=/dir/to/caffe/python:$PYTHONPATH

/dir/to/caffe/python 是 caffe 目录下的 python 目录，针对实际情况进行修改即可。

3. 更新环境变量配置

ldconfig

有时可能需要关闭当前会话再启动会话以生效。

4. 编译 python 接口

cd ~/caffe/

make pycaffe

出现错误提示时，通常是 Makefile.config 里面的 PYTHON_INCLUDE 或者 PYTHON_LIBRARIES配置有误，根据错误提示进行修改，重新编译即可。

5. 验证是否安装成功

在 python console 里执行 import caffe，无误则表示 caffe python 接口安装成功；

5. 调用 caffe 的 C++ 接口

这部分内容目前还在学习中，希望可以早日学会，分享进步。