TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Machine Learning Model TrainingML FrameworksOpen Source

About

Torch-TensorRT intègre TensorRT avec PyTorch pour améliorer les performances d'inférence sur les plateformes NVIDIA. Cela te permet d'accélérer considérablement l'inférence des modèles, atteignant des temps d'exécution plus rapides avec des modifications minimales du code. L'outil prend en charge les environnements Python et C++, te permettant d'optimiser et de déployer tes modèles de manière efficace. Ce produit convient aux développeurs travaillant avec des applications d'apprentissage profond qui nécessitent une performance améliorée sur le matériel NVIDIA.

Open Source Health

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

Resources & Links

Related Categories

Tu utilises TensorRT ?

Suis son coût avec le reste de ton stack et reçois un rappel avant son renouvellement.

Ajouter à mon stack

Vendor

PyTorch

Publisher of Captum and Bayesian Optimization Library (botorch)

View vendor profile

More by PyTorch