[人工智能]Pytorch

环境搭建

安装Python环境依赖

安装Python环境依赖：
pip install graphviz  # 安装graphviz
pip install git+https://github.com/szagoruyko/pytorchviz  # 通过git安装torchviz

安装系统graphviz 运行环境

下载地址：graphviz.gitlab.io/_pages/Download/Download_windows.html

使用方法

1. 加载并运行待可视化模型

import torch
from torchvision.models import AlexNet
from torchviz import make_dot
import os
os.environ["PATH"] += os.pathsep + 'D:/Program Files (x86)/Graphviz2.38/bin/'

x = torch.rand(8, 3, 256, 512)
model = AlexNet()
y = model(x)

2. 调用make_dot()函数构造图对象

#三种方式选择其一即可
g = make_dot(y)
g = make_dot(y, params=dict(model.named_parameters()))
g = make_dot(y, params=dict(list(model.named_parameters()) + [('x', x)]))

3. 保存图结构PDF

# 这两种方法都可以
# g.view() # 会生成一个 Digraph.gv.pdf 的PDF文件
g.render('espnet_model', view=False) # 会自动保存为一个 espnet.pdf，第二个参数为True,则会自动打开该PDF文件，为False则不打开

【注】

一定要加载运行时系统环境变量，

import os
os.environ["PATH"] += os.pathsep + 'D:/Program Files (x86)/Graphviz2.38/bin/'

不然会报错

graphviz.backend.ExecutableNotFound: failed to execute 'dot', make sure the Graphviz executables are on your systems' PATH