T
torch2trt
NVIDIA-AI-IOT/torch2trt
An easy to use PyTorch to TensorRT converter
★4.9kstars
Python
MIT
Updated: Today
📋 Project at a Glance
Tap to expand
What's this?A open-source Data & Infrastructure project, built with Python, focusing on classification and inference
Who made it?Maintained by NVIDIA-AI-IOT team, 4.9K⭐ on GitHub, #376 out of 3133 in Data & Infrastructure
Why does it exist?The NVIDIA-AI-IOT team recognized that existing classification tools in Data & Infrastructure were hard to use. torch2trt was designed to make inference more accessible.
What can it do?Key use cases: jetson-nano, jetson-tx2, jetson-xavier
How to install with AI?Use an AI coding assistant (Claude Code, Cursor, Copilot) to automatically set up pip dependencies and virtual env. Follow the README — the AI handles the rest.
🔗 github.com/NVIDIA-AI-IOT/torch2trt
🔗 github.com/NVIDIA-AI-IOT/torch2trt
Topics
classificationinferencejetson-nanojetson-tx2jetson-xavierpytorchtensorrt