配置caffe的python接口的步骤总结:
a. 安装依赖库
库的版本要求,在$caffe-root/python中的requirements.txt
Cython>=0.19.2
numpy>=1.7.1
scipy>=0.13.2
scikit-image>=0.9.3
scikit-learn>=0.14.1
matplotlib>=1.3.1
ipython>=1.1.0
h5py>=2.2.0
leveldb>=0.191
networkx>=1.8.1
nose>=1.3.0
pandas>=0.12.0
python-dateutil>=1.4,<2
protobuf>=2.5.0
python-gflags>=2.0
pyyaml>=3.10
Pillow>=2.7.0
b. 编译 make pycaffe
c. 把caffe的python模块接入PYTHONPATH,
cd
sudogedit ~/.bashrc
加入exportPYTHONPATH=/path/to/caffe/python:$PYTHONPATH
d. 测试
python
>>import caffe
易错点一:装完依赖项之后,忘记编译了,即忘了执行make pycaffe。所以,import caffe时就会提示找不到_caffe。
易错点二:make pycaffe 时需要修改Makefile.config,主要是各种路径的修改,如果Makefile.config没改好,编译的时候就会报错。
分析:报错提示,找不到Python.h,其实就是各种路径的设置出错。
其实,解决方案很简单,在Makefile.config里面都有提示你去如何设置路径。想执行make pycaffe。需要在Makefile.config改动4个地方。1、
分析:添加python相应的路径,使其能找到Python.h。以及numpy/arrayobject.h. 检查我的设置是否符合要求:
满足条件
======================================================================
2、
3、
4、
总结:必须要在Makefile.config进行上述4项改动,才能够执行make pycaffe。
使caffe的python接口永久生效。
vim /etc/profile
exportPYTHONPATH=/mnt/caffe/python:$PYTHONPATH
source /etc/profile
配置成功: