【caffe】ubuntu配置python接口----pycaffe

1.安装python(系统默认自带python27,若需要安装 :sudo apt-get install python-dev)

2.安装python依赖,进入到caffe/python中,打开命令窗,输入命令:

for req in $(cat requirements.txt); do sudo pip install $req; done

3.修改makefile.config文件

PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/local/lib/python2.7/dist-packages/numpy/core/include

4.添加路径
sudo gedit /etc/profile
在末尾添加:

export PYTHONPATH=xxx/caffe/python:$PYTHONPATH

(注意 xxx/caffe,的路径)

5.重新编译
sudo make clean
sudo make all -j4
sudo make pycaffe
进入python , import caffe ,成功

补充一个可能的大坑:

在运行py文件进行测试的时候,如果出现

F1122 13:37:10.906644 32369 syncedmem.hpp:22] Check failed: error == cudaSuccess (30 vs. 0)  unknown error
*** Check failure stack trace: ***
Aborted (core dumped)

请采用 sudo python xxx.py