ubuntu16.04配置caffe(CPU版)

来源:互联网 发布:js实现四张图片轮播 编辑:程序博客网 时间:2024/06/04 17:58

参考博文:http://blog.csdn.net/u013989576/article/details/56677749

一、安装依赖项
把下面四个命令依次复制运行即可;

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

二、下载caffe源码
没有安装git的话需要先装一下git

sudo apt-get install git

下载Caffe源码:

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

三、编译caffe

1)进入caffe目录:cd caffe/

2)生成Makefile.config文件:cp Makefile.config.example Makefile.config

3)修改Makefile.config文件中的配置

a) 编辑Makefile.config文件:sudo gedit Makefile.config
b)去掉CPU_ONLY前面的#号
c) 配置引用文件路径(主要是HDF5的路径问题:hdf5.h、 libhdf5.so的路径,可以利用 sudo find / -name 【文件名】来查找某个文件的具体位置)

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include/usr/include/hdf5/serial (查找hdf5.h路径:sudo find /-name hdf5.h)

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib/usr/lib/i386-linux-gnu/hdf5/serial (查找libhdf5.so路径:sudo find /-name libhdf5.so)

d) 执行编译

sudo make all -j4sudo make test -j4sudo make runtest -j4

执行完runtest后,如果显示PASSED,则执行成功;
注意:如果执行结果失败,则需要执行语句sudo make clean后,解决掉问题,重新编译

四、编译python接口

1) 安装pip

命令如下:sudo apt-get install python-pip

2) 安装Python接口依赖库

在caffe根目录下,有个python文件夹,该文件夹有requirements.txt,里面有需要的依赖库和版本信息,按照其安装即可:

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

安装结束后,可以执行如下语句验证:sudo pip install -r requirements.txt,如果显示Requirement already satisfied,则安装成功,否则会继续进行安装。

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

a) 打开配置文件bashrc:sudo gedit ~/.bashrc

b) 在文件的最后面添加:

export PYTHONPATH=/home/hanyuying/caffe/python:$PYTHONPATH(进入caffe文件夹里的python目录下,输入pwd就可以查看路径)

4)编译python接口

cd ~/caffe/

sudo make pycaffe

注意:如果出现找不到numpy/arrayobject.h这种问题,则检查Makefile.config文件中的PYTHON路径(Python.h、numpy/arrayobject.h的路径):

PYTHON_INCLUDE := /usr/include/python2.7 /usr/lib/python2.7/dist-packages/numpy/core/include

5) 验证python接口

进行python环境,引入caffe包:import caffe,如果没有报错,则安装成功!

原创粉丝点击