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 scipyeasy_install protobuf
安装完成上述caffe依赖的python库后,可能需要重新执行一遍第4,5步操作。