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