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

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”文件