torchviz

A small package to create visualizations of PyTorch execution graphs

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install torchviz

Dependencies

Releases

Version Released Bookworm
Python 3.11
Trixie
Python 3.13
Files
0.0.3 2024-12-02    
0.0.2 2021-03-26
0.0.1 2019-01-28

Issues with this package?

Page last updated 2026-09-03 10:00:36 UTC