环境:ubuntu14.04 python2.7 caffe已经成功编译
1,首先确保pip已经安装
sudo apt-get install python-pip
2,在caffe-master/python 下有一个requirement.txt,里面有所需要安装的所有库。
cd caffe-master/python for req in $(cat requirements.txt); do pip install $req; done
执行上述代码,可以自动安装所需要的库。
但我在第一次执行时出错了,说是好多引用找不到。后来,打开了sudo gedit /etc/profile 在后面加了个几个路径,再执行就可以正常安装了,路径如下:
export PYTHONPATH=${HOME}/caffe-master/python:$PYTHONPATH
export LD_LIBRARY_PATH=${HOME}/caffe/build/lib:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
路径方面有些不太明白,这里具体也不知道是怎么回事。
如果执行成功,就代表所需要的库已经安装成功。可以在进入 caffe-master/python 路径下执行:sudo pip install -r python/requirements.txt
就会看到,安装成功的,都会显示Requirement already satisfied, 没有安装成功的,会继续安装。
3,pycaffe编译
cd caffe-master
make pycaffe
4,加入环境变量
sudo gedit /etc/profile
删除之前加入的路径,在后面加入,export PYTHONPATH=~/caffe-master/python:$PYTHONPATH
source /etc/profile 应用路径
5,验证
python
import caffe
如果不报错,就是编译成功了。
PS:1,不知道为什么,每次新打开一个终端,都要重新 source /etc/profile ,否则在python下导入caffe会报错。
2,之前要对 Makefile.config修改
WITH_PYTHON_LAYER := 1 #取消注释