ubuntu20下使用 torchviz可视化计算图

1. 安装 torchviz：

pip install torchviz

2. 示例代码：下面是一个简单的示例代码，展示如何使用 torchviz 可视化计算图：

python
import torch
from torchviz import make_dot

# 创建一个简单的模型
model = torch.nn.Sequential(
    torch.nn.Linear(2, 2),
    torch.nn.ReLU(),
    torch.nn.Linear(2, 1)