本文介绍了在Windows环境中，如何在没有Anaconda的情况下使用Graphviz和torchviz来可视化Pytorch的网络结构。首先，详细阐述了Graphviz的下载、安装及环境变量配置步骤。接着，说明了torchviz的安装过程。通过代码示例展示了如何生成并查看网络结构图。在遇到未自动生成PDF文件的问题时，提供了手动转换gv文件为图片的解决方法，并给出了相关错误处理的提示。
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(
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
