1.配置python
在安裝caffe時未配置python,現在需要配置caffe文件中的Makefile.config文件,配置python的安裝路徑(默認即可),配置python依賴包:
sudo apt-get install python-pip
再轉到caffe的python目錄,安裝scipy
cd caffe/python sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose sudo apt-get install python-numpy python-scipypython-matplotlib python-sklearn python-skimage python-h5py python-protobufpython-leveldb python-networkx python-nose python-pandas python-gflags Cythonipython
最後安裝requirement裏面的包,需要root權限
sudo su for req in $(cat requirements.txt); do pip install $req; done回到caffe目錄,然後編譯
<span style="font-size:18px;">make -j
make pycaffe</span>
檢測是否成功配置python<span style="font-size:18px;">python
import caffe</span> //若無錯誤則安裝成功,進行下面的步驟
2.安裝pydot
python的pydot模塊是Graphviz的dot語言的python接口,主要使用Graphviz繪圖,
sudo apt-get update
sudo apt-get install python-pydot
3.安裝Graphvizsudo apt-get install graphviz
sudo apt-get update
4.繪製net
切換到caffe安裝目錄中的Python文件夾,檢查是否存在draw_net.py文件,若存在(不存在說明前面出現安裝過程錯誤)則執行命令
python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png //lenet.png就是繪製的圖形