Caffe Python接口编译

1.安装python:

yum install python-devel.x86_64

2.Makefile.config修改

WITH_PYTHON_LAYER := 1 #取消注释

3.安装python依赖

如果安装 Anaconda Python可以免去下面步骤

cd $CAFFE_ROOT/python
for req in $(cat requirements.txt); do pip install $req; done

4.编译

make clean
make pycaffe -j32
make test -j32

 5. 使Caffe的python接口永久生效

vim /etc/profile
profile最后添加: export PYTHONPATH=/root/$CAFFE_ROOT/python:$PYTHONPATH
source /etc/profile

注意:

有一些python库没有安装,在import caffe测试时会出错,常见的库缺失错误:

ImportError: No module named skimage.io

ImportError:  No module named scipy

ImportError:  No module named google.protobuf.internal

yum install python-matplotlib.x86_64
pip install -U scikit-image
pip install scipy
easy_install protobuf

安装完成上述caffe依赖的python库后,可能需要重新执行一遍第4,5步操作。

posted @ 2016-07-12 16:48  venus024  阅读(6433)  评论(0编辑  收藏  举报