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

时间:2021-08-24 06:30:17

在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