Pytorch模型可视化结构——超级方便好用的工具Torchviz及failed to execute [‘dot‘, ＞ ‘-Tpdf‘, ‘-O‘, ‘Digraph.gv‘]问题解决

推荐一个非常好用的可视化工具，torchviz/graphviz

第一步，下载

首先要下载Graphviz， [官网地址]，我是在windows下，所以直接下载windows版本的zip就可以，下载后找个地方解压就行(http://www.graphviz.org/download/)

第二步，配置Graphviz的系统环境变量

在高级系统设置—>系统属性—>高级—>环境变量—>系统变量—>Path中添加Graphviz/bin的路径，我是D:\YXC\Study\Graphviz\Graphviz\bin

验证是否配置成功：
cmd控制台输入 dot -version，如果显示下面就没问题了！

第三步，python环境中install

pip install torchviz
pip install graphviz

最后一步，重启编译器

用法

from torchviz import make_dot
import XXX
net = your_net()
net_input = XXX(定义一个与网络输入维度相同的变量）
out = net(net_input)
vis_graph =make_dot(out,params=dict(net.named_parameters()))
vis_graph.view()

好了，到这里就可以直观的看到网络结构了

问题1

graphviz.backend.ExecutableNotFound: failed to execute [‘dot’,
‘-Tpdf’, ‘-O’, ‘Digraph.gv’], make sure the Graphviz executables are
on your systems’ PATH

这个有两个原因，一种是没有把下载的Graph/bin添加到环境变量中，另一种是没有重启编译器，按照上面的步骤来的话就没有问题了。