Pytorch 模型的网络结构可视化 pytorchviz

一、安装命令：

sudo pip install graphviz #

或

sudo pip install git+https://github.com/szagoruyko/pytorchviz

二、报错问题

使用graphviz 生成树报错(graphviz.backend.ExecutableNotFound: failed to execute ['dot', '-Tpdf', '-O', 'Digraph.gv'])

解决方法：
首先，在官网上下载软件，安装完成后，手动配置环境变量或者安装时，勾选添加进系统环境变量中。
  官网地址：http://www.graphviz.org/download/   （https://www2.graphviz.org/Packages/development/windows/10/cmake/Release/x64/）

其次，在python terminal内通过pip install graphviz 安装组件。

最后，重启一下编译器（pycharm）后，就可以正常使用了

三、实验代码

扫描二维码关注公众号，回复： 13109785 查看本文章

x = torch.randn(1,8)

vis_graph = make_dot(model(x), params=dict(model.named_parameters()))

vise_graph.view()

结果：

参考:

[1] :  Python库 - Pytorch 模型的网络结构可视化 pytorchviz -

[2]：使用graphviz 生成树报错(graphviz.backend.ExecutableNotFound: failed to execute ['dot', '-Tpdf', '-O', 'Digraph.gv'])