Ways to Create computational graph image?

vision
1

I would like to know about the ways to create a computational graph image from PyTorch code?

One way is using the SummaryWriter in Tensorboard.
Is there any other way to create this? Are any other packages/libraries available?

2

torchviz can also be used to visualize the computation graphs.