1、安装python所需的依赖包
(1)sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose
(2)再使用pip安装requirements.txt文件中所指定的包(若之前没安装过pip,需要安装pip:sudo apt-get install python-pip)
在caffe根目录下使用命令:
for req in $(cat python/requirements.txt); do pip install $req; done
(在这之前需要使用命令su root转到root用户,因为安装过程需要root权限)
2、Makefile.config文件中,将
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
改为
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/local/lib/python2.7/dist-packages/numpy/core/include
3、进行编译
make pycaffe
(如果编译过程出错,则可能是依赖包没装好,我之前出现过protobuf找不到的错误,此时用pip install protobuf安装重新安装一下就好)
4、验证
进入python文件夹:cd python
使用命令:python
import caffe
(如果没有出错,则说明安装好了)
5、使用draw_net.py绘制lenet网络结构图
命令:python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png
此时出现错误:ImportError: No module named pydot,则再次使用pip安装,pip install pydot
再次运行:python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png
又出现错误了:pydot.InvocationException: GraphViz's executables not found,坑爹。此时需要安装graphviz,命令为sudo apt-get install graphviz
最后再运行一次,终于成功了。在python文件夹下得到lenet.png文件了。如下图: