Visualize TensorRT engine's graph

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

Description

Is there a tool to visualize a TensorRT engine graph? I’m used to visualizing models with Netron, but it doesn’t support TensorRT. Is there a way Nvidia supports visualizing TensorRT models?

Environment

TensorRT Version : 7.1.2
CUDA Version : 11.0
Operating System + Version : Ubuntu 18.04
Python Version (if applicable) : 3.6

1 Like
3

Hi @weissrael,

We don’t have such a tool available for visualizing the TRT engine. In case you’re using tf-trt, you can visualize tf-trt optimized graph using tensorboard.

Thank you.