TensorRT
Machine Learning Model TrainingML Frameworks

TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Open Source

About

Torch-TensorRT integrates TensorRT with PyTorch to enhance inference performance on NVIDIA platforms. It allows users to accelerate model inference significantly, achieving faster execution times with minimal code changes. The tool supports both Python and C++ environments, enabling users to optimize and deploy their models efficiently. This product is suitable for developers working with deep learning applications who require improved performance on NVIDIA hardware.

Open Source Health

Not enough history
Stars
2,993
Forks
410
License
BSD-3-Clause
Last commit
26 days ago
Python

Related Categories