pytorch模型可视化：pytorchviz安装及使用

原创已于 2025-04-23 09:09:59 修改 · 6.4k 阅读

17 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#pytorch #python

于 2023-02-11 20:32:35 首次发布

文章介绍了如何安装和使用PyTorchViz，这是一个用于PyTorch模型可视化的工具。首先需要安装graphviz，然后通过pip安装PyTorchViz，可能需要解决git安装和URL替换的问题。安装完成后，通过make_dot函数可以可视化如AlexNet等网络结构。

pytorch模型可视化：pytorchviz安装及使用

pytorchviz是pytorch常用的模型可视化工具，下面介绍一下pytorchviz的安装及使用过程。

1.命令行安装pytorchviz(安装之前需要先安装graphviz)

pip install graphviz(如果之前下载了无需安装)
pip install git+https://github.com/szagoruyko/pytorchviz

2.如果执行命令pip install git+https://github.com/szagoruyko/pytorchviz报下面错误：

在这里插入图片描述

3.需在执行命令前先安装git:

conda install git

4.安装完git再执行上面那条命令如果报错：

在这里插入图片描述
则将命令修改：

原安装命令： pip install git+https://github.com/szagoruyko/pytorchviz
修改后的安装命令： pip install git+git://github.com/szagoruyko/pytorchviz

5.如果出现下面警告：

在这里插入图片描述
则使用命令：

git config --global url."https://github.com".insteadOf git://github.com
再运行：pip install git+git://github.com/szagoruyko/pytorchviz

6.安装成功后使用torchviz可视化torch网络结构，以alexnet为例。

import torch
import torchvision.models as models
from torchviz import make_dot

alexnet=models.alexnet().to(device)
print(alexnet)
g=make_dot(alexnet(torch.rand(1,3,64,64)),params=dict(model.named_parameters()))
g.view()# 会在当前目录下保存一个“Digraph.gv.pdf”文件

在这里插入图片描述