Repolode
NVIDIA

NVIDIA/TensorRT

Открытые компоненты NVIDIA TensorRT — SDK для высокопроизводительного инференса моделей глубокого обучения на GPU NVIDIA: плагины, ONNX-парсер и примеры.

Звёзды
13,411
Форки
2,412
Открытые issues
668
Последний коммит
13 дней назад

Что это?

TensorRT — это SDK NVIDIA для инференса моделей глубокого обучения на GPU NVIDIA. В этом репозитории находится открытая часть релиза TensorRT: исходный код плагинов и ONNX-парсера, а также примеры приложений, показывающие, как пользоваться платформой. Runtime и библиотека с закрытым исходным кодом поставляются в составе релиза general availability, под который собирается этот репозиторий.

Проект поддерживает сборку для Linux на x86-64 и aarch64 (включая кросс-компиляцию для Jetson и DriveOS), а также для Windows. В качестве системы сборки используется CMake. Для Linux доступен контейнеризированный процесс сборки на базе Docker. Пакет pip tensorrt позволяет установить Python bindings без сборки из исходного кода.

В репозитории также описаны пути импорта моделей, в том числе ONNX, Torch-TensorRT, HuggingFace/Optimum и Network Definition API, а также приведена матрица поддержки отдельных моделей: LLM, NLP-энкодеров, моделей для компьютерного зрения, аудио, диффузионных и мультимодальных моделей.

Ключевые возможности

  • 01Исходный код открытых плагинов и ONNX-парсера для TensorRT.
  • 02Примеры приложений, демонстрирующие использование TensorRT.
  • 03Сборка для Linux (x86-64, aarch64) и Windows с помощью CMake.
  • 04Контейнеры Docker для воспроизводимой сборки под Linux.
  • 05Python-пакет, который можно установить через pip.
  • 06Инструментарий для кросс-компиляции под Jetson и DriveOS.

Сценарии использования

  • Развёртывание моделей глубокого обучения для инференса с низкой задержкой на GPU NVIDIA.
  • Преобразование моделей ONNX или PyTorch в оптимизированные TensorRT engines.
  • Создание собственных плагинов TensorRT для слоёв, которые не поддерживаются из коробки.
  • Запуск инференса на встраиваемых устройствах Jetson или DriveOS.

Быстрый старт

  1. 1

    Установите предварительно собранный пакет TensorRT для Python

    pip install tensorrt

Что важно знать

  • Требуется графический процессор NVIDIA с установленной CUDA.
  • Репозиторий с открытым исходным кодом представляет собой подмножество релиза TensorRT GA; для полноценной нативной сборки потребуется отдельная сборка TensorRT.
  • Для сборки из исходного кода требуются значительные системные зависимости, такие как CUDA, cuDNN и библиотеки TensorRT GA.

Темы

Частые вопросы

Что такое TensorRT?
Открытые компоненты NVIDIA TensorRT — SDK для высокопроизводительного инференса моделей глубокого обучения на GPU NVIDIA: плагины, ONNX-парсер и примеры. TensorRT — это SDK NVIDIA для инференса моделей глубокого обучения на GPU NVIDIA. В этом репозитории находится открытая часть релиза TensorRT: исходный код плагинов и ONNX-парсера, а также примеры приложений, показывающие, как пользоваться платформой. Runtime и библиотека с закрытым исходным кодом поставляются в составе релиза general availability, под который собирается этот репозиторий.
TensorRT можно использовать бесплатно? Какая у него лицензия?
TensorRT распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
TensorRT всё ещё поддерживается?
Активно поддерживается. Последний коммит в TensorRT — 9 окт. 2026 г..
Для чего можно использовать TensorRT?
Развёртывание моделей глубокого обучения для инференса с низкой задержкой на GPU NVIDIA. Преобразование моделей ONNX или PyTorch в оптимизированные TensorRT engines. Создание собственных плагинов TensorRT для слоёв, которые не поддерживаются из коробки. Запуск инференса на встраиваемых устройствах Jetson или DriveOS.
Как установить TensorRT?
Установите предварительно собранный пакет TensorRT для Python: pip install tensorrt
Где скачать TensorRT?
Исходный код и инструкции по установке — на GitHub: https://github.com/NVIDIA/TensorRT

Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.

Похожие проекты

Фреймворк для развёртывания ИИ с визуальным редактором рабочих процессов и поддержкой запуска моделей на настольных, мобильных, периферийных и серверных устройствах через разные движки инференса.

1,9 тыс.231C++Apache-2.0Обновлён 2 месяца назад