Windows环境下 Pytorch结合Graphviz+torchviz 进行网络结构可视化（无Anaconda环境）

1、安装Graphviz

（1）进入https://www2.graphviz.org/Packages/stable/windows/10/cmake/Release/x64/，下载exe文件，如下图所示。

下载后打开安装，安装路径可以自己选择。安装完毕后，将Graphviz安装路径下的bin文件目录加入到环境变量中去。

右击“此电脑”-“属性”-“高级系统设置”-“环境变量”-在上方“用户变量”的Path栏增加bin目录路径，如“D:\software\Graphviz\bin”

（2）打开cmd命令行，运行

pip install graphviz

完成之后需要重启，重启后在cmd下输入

dot -version

有版本信息显示说明安装成功。 

2、安装torchviz

pip install torchviz

 

3、代码部分

from torchviz import make_dot

out = model(data_x) #model为继承nn.Module自定义建立的模型
g = make_dot(out)
g.view("test-output/test-table.gv")

自动创建的test-output文件夹中包含两个文件“test-table.gv”和“test-table.gv.pdf”

4、可能出现的错误：

（1）如果没有自动创建pdf文件，可以利用命令行将.gv文件转换为图片。

dot target.gv -Tpng -o target.png

（2）在cmd下输入“dot -version”后显示以下信息:

Perhaps "dot -c" needs to be run (with installer's privileges) to register the plugins?

执行dot -c命令即可。

参考

https://blog.csdn.net/u010626747/article/details/106802150

https://www.jianshu.com/p/3a37a373b8a0