在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