公司服务器 caffe 绘制网络结构图 及 问题解决

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

完美运行,无错误报出~

发布了39 篇原创文章 · 获赞 12 · 访问量 16万+

猜你喜欢

转载自blog.csdn.net/wdh315172/article/details/79127979
今日推荐