1、安装pydot
$ sudo apt-get install python-pydot
2、安装graphviz
$ sudo apt-get install graphviz
3、生成
python draw_net.py ../examples/cifar10/cifar10_quick_train_test.prototxt cifa10.png
/home/ubuntu/caffe/python/cifa10.png
python draw_net.py ../examples/cifar10/cifar10_quick_train_test.prototxt ../models/model_picture_s/cifa10.png
Drawing net to ../models/model_picture_s/cifa10.png
结果: