您的位置:首页 > 编程语言 > Python开发

在Ubuntu16.04上安装Caffe(无GPU),并配置python接口

2017-03-27 19:37 471 查看
在Ubuntu16.04上安装Caffe,并配置python接口,主要参考:

http://blog.csdn.net/muzilinxi90/article/details/53673184#reply

http://blog.csdn.net/muzilinxi90/article/details/53673184#reply

http://caffe.berkeleyvision.org/installation.html

**

1.安装依赖库

sudo apt-get install libprotobuf-dev
sudo apt-get install libleveldb-dev
sudo apt-get install libsnappy-dev
sudo apt-get install libopencv-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev


安装BLAS

sudo apt-get install libatlas-base-dev


由于系统中自带python,所以我们使用默认的python来建立pycaffe接口,需要安装

sudo apt-get install python-dev


以及一些兼容性的依赖库

apt-get install libgflags-dev
apt-get install libgoogle-glog-dev
apt-get install liblmdb-dev


**

2.下载caffe

**

首先安装git

sudo apt-get install git


下载caffe源码

可以直接在网页下载 https://github.com/BVLC/caffe.git

也可输入命令

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


切换到caffe下的python目录下,安装python依赖库

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


3.编译caffe

到Caffe文件夹中,拷贝一份Makefile.config.example并重命名成Makefile.config,修改该配置文件:

cp Makefile.config.example Makefile.config


vim Makefile.config


在编辑器中,去掉CPU_ONLY:=1前面的注释,表明我们使用的是CPU;由于Ubuntu16.04文件结构的变化,#Whatever else you find you need goes here.处要改成下面这样:

# 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 pycaffe


在这里 我出现了

python/caffe/_caffe.cpp:10:31: fatal error: numpy/arrayobject.h: 没有那个文件或目录

compilation terminated.

Makefile:507: recipe for target ‘python/caffe/_caffe.so’ failed

make: * [python/caffe/_caffe.so] Error 1

这样的错误,于是我执行了以下方案

sudo apt-get install python-numpy


sudo make pycaffe -j16


然后

make clean


接下来进行编译

make pycaffe
make all
make test
make runtest


make 默认单核运算,若处理器是4核,在每条命令后加上-j4,可以加快速度

**

4.测试

**

测试Caffe的Python接口,切换到caffe/python文件目录下,记录下来当前路径,输入以下命令:

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


在命令行输入

python


进入python环境

输入

import caffe


如果没有报错,证明安装成功。

上面的方法,一旦关闭终端或者打开新终端则失效,如果放到配置文件中,可以永久有效果,命令操作如下:

#A.把环境变量路径放到 ~/.bashrc文件中
sudo echo export PYTHONPATH="~/caffe/python" >> ~/.bashrc
#B.使环境变量生效
source ~/.bashrc


最近试了一种方法,感觉挺方便

还有,其实可以也安装pycharm 然后每次写程序的时候 加上

import sys

sys.path.append(‘/path/to/caffe/python’)

import caffe

也可以非常方便的使用caffe
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  ubuntu python