pytorch网络可视化

安装

1. 首先确保python环境有pytorch和torchvision
2. 在环境中安装torchviz：先切到环境中，然后键入 pip install torchviz
3. 在系统中安装第三方软件graphviz：
   ○ linux：终端键入 sudo apt install graphviz
   ○ windows：去官网下载安装包，安装的时候勾选加入环境变量PATH的选项，否则你需要手动设置把graphviz的bin目录放到path里
4. 检查graphviz安装情况：打开终端/cmd，键入dot --version，正常打印版本信息说明安装成功

使用

运行以下测试代码（resnet18）

import torch
from torchvision import models
from torchviz import make_dot
models = models.resnet18