1. 在caffe-master路径下直接运行【在成功编译caffe的源码之后,可以在python环境中使用caffe】:
python ./python/draw_net.py ./models/bvlc_alexnet/train_val.prototxt ./AlexNet.jpg--rankdir=LR
参考:https://www.cnblogs.com/denny402/p/5106764.html
2. 但是在运行的时候出现了这样的错误,
ImportError: libnccl.so.1: cannot open shared object file: No such file or directory
解决方案:
(1)参考: http://blog.csdn.net/liuweizj12/article/details/52138191 未解决;
(2)参考:http://blog.csdn.net/qq_15309757/article/details/71244346 ,发现相似的问题:
(3)接着参考红色链接:http://www.cnblogs.com/Anker/p/3209876.html
公司的共享文件 属于 情况2)。
cat /etc/ld.so.conf
# include ld.so.conf.d/*.conf
# /usr/local/cuda-8.0/lib64
echo '/usr/local/lib' >> /etc/ld.so.conf
ldconfig # 在 /root 路径下
3. 再次运行:
python ./python/draw_net.py ./models/bvlc_alexnet/train_val.prototxt ./AlexNet.jpg--rankdir=LR
运行成功,产生结果。
不过报出个小问题:
Couldn't import dot_parser, loading of dot files will not be possible.
解决:
(1)pip install pydotplus
(2)cd /usr/lib/python2.7/site-packages
(3)vim dot_parser.py
修改第19行,将 import pydot 改成 import pydotplus as pydot
解决方案参考:http://blog.csdn.net/wwsqt/article/details/66971789
4. 再次运行:
python ./python/draw_net.py ./models/bvlc_alexnet/train_val.prototxt ./AlexNet.jpg--rankdir=LR
完美运行,无错误报出~