本文介绍PyTorch可视化工具库PyTorchViz的下载与安装以及测试代码。
一、下载与安装
1.1 下载
下载地址:https://github.com/szagoruyko/pytorchviz
1.2 安装
首先安装graphviz
pip install graphviz
然后安装torchviz:
pip install torchviz
1.3 单独安装graphviz
Windows系统下,还得单独安装软件graphviz,下载地址:https://graphviz.org/download/,选择64位的exe软件。
双击exe进行安装;
在此过程中需要注意的就这一步,可选择是否自动添加环境变量,建议直接选第二个:
测试是否安装成功;
Win+R打开命令行窗口,输入以下命令
dot -version
若出现如下界面,则证明安装成功。
二、测试
测试代码如下: