Caffe Python接口编译

来源:互联网 发布:2017网络用语及解释 编辑:程序博客网 时间:2024/04/28 07:39

1.安装python:

yum install python-devel.x86_64

2.Makefile.config修改

WITH_PYTHON_LAYER := 1 #取消注释

3.安装python依赖

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

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

4.编译

make cleanmake pycaffe -j32make test -j32

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

vim /etc/profileprofile最后添加: export PYTHONPATH=/root/$CAFFE_ROOT/python:$PYTHONPATHsource /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_64pip install -U scikit-image
pip install scipy
easy_install protobuf

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

原创粉丝点击