您的位置:首页 > 理论基础 > 计算机网络

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

2018-01-22 13:32 429 查看
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

完美运行,无错误报出~
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  caffe