配置caffe的python接口及其易错点

配置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. caffepython模块接入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

配置成功:



没有更多推荐了,返回首页