Background 背景: 当我们需要使用Python来调用caffe 的时候,如import caffe,则我们需要配置一下PyCaffe
Environment 环境:Ubuntu16.04 on VMware Workstation、Intel Caffe
0. 安装Caffe
可参考之前的文章
1. 安装依赖库
sudo apt-get update
sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags Cython ipython
sudo apt-get install protobuf-c-compiler protobuf-compiler
sudo pip install -r ${CAFFE_ROOT}/python/requirements.txt
这里还需要将${CAFFE_ROOT}替换成caffe的安装路径
sudo pip install pydot
2. 编译
cd caffe
make pycaffe
之后做一个简单的测试,进入caffe/python下,启动python
3. 环境配置
sudo gedit /etc/profile
添加:export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
source /etc/profile
使之生效
4. 完成
此时就可以直接用pycaffe了