trt-onnx-quickstart

Official

Convert ONNX models to TensorRT for efficient inference.

AuthorNVIDIA
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill enables developers to convert ONNX models to TensorRT, ensuring efficient and accurate inference on NVIDIA GPUs.

Core Features & Use Cases

  • ONNX to TensorRT Conversion: Build and verify TensorRT engines from ONNX models for high-performance inference.
  • Numerical Parity Check: Ensures that TensorRT engines match the outputs of ONNX Runtime within specified tolerances.
  • Dynamic Shapes and Multi-Input Support: Handles multi-input models and dynamic shapes, including batch size and sequence length variations.

Quick Start

Run the trt-onnx-quickstart skill with the Hugging Face model ID or ONNX file to generate a TensorRT engine and verify its numerical parity against ONNX Runtime.

Dependency Matrix

Required Modules

tensorrtpythonpolygraphyonnxruntimeonnxsimonnx_graphsurgeonoptimum-onnxsentence-transformers

Components

scriptsreferences

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: trt-onnx-quickstart
Download link: https://github.com/NVIDIA/TensorRT/archive/main.zip#trt-onnx-quickstart

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 620,000+ vetted skills library on demand.