torchviz工具入门

一、定义

1. 定义
2. 部署
3. 案例

二、实现

1. 定义
   TorchViz是一个用于可视化 PyTorch计算图的工具库，查看模型结构、以及算子流向图。
   文档：https://graphviz.org/
2. 部署
   1. 安装torchviz

pip install torchviz
pip install graphviz

2. 安装graphviz.exe
   下载:https://graphviz.gitlab.io/_pages/Download/Download_windows.html
   
   配置：
   
   
   
   

打开cmd(win+R,输入cmd),或者打开Anaconda Promopt
输入dot -v，如果呈现如下的结果，就证明配置成功

运行报错：graphviz.backend.execute.calledprocesserror: command '[‘dot’, ‘-kdot’, ‘-tpdf’, ‘-o’, ‘mynetmodel’
解决：

1.打开cmd
2.运行：dot -v
3.出现问题：There is no layout engine support for “dot”…
4.运行：dot -c
5.没有提示（成功）
6.再次运行dot -v

3. 案例

import torch
from torchviz import make_dot
from torchvision.models import vgg16  # 以 vgg16 为例

x = torch.randn(4, 3, 32, 32)  # 随机生成一个张量
model = vgg16()  # 实例化 vgg16，网络可以改成自己的网络
out = model(x)  # 将 x 输入网络
g = make_dot(out)  # 实例化 make_dot
# g.view()  # 直接在当前路径下保存 pdf 并打开
g.render(filename='111', view=False, format='pdf')  # 保存 pdf 到指定路径不打