Yolo Vision Shenzhen
Shenzhen
Şimdi katılın
Sözlük

TensorRT

NVIDIA GPU'larda daha hızlı ve verimli çıkarım için derin öğrenme modellerini TensorRT ile optimize edin. YOLO ve yapay zeka uygulamalarıyla gerçek zamanlı performans elde edin.

TensorRT is a high-performance deep learning inference software development kit (SDK) developed by NVIDIA. It is designed to optimize neural network models for deployment, delivering low inference latency and high throughput for deep learning applications. By acting as an optimization compiler, TensorRT takes trained networks from popular frameworks like PyTorch and TensorFlow and restructures them to execute efficiently on NVIDIA GPUs. This capability is crucial for running complex AI models in production environments where speed and efficiency are paramount.

How TensorRT Optimizes Models

The core function of TensorRT is to convert a trained neural network into an optimized "engine" specifically tuned for the target hardware. It achieves this through several advanced techniques:

  • Layer Fusion: The optimizer combines multiple layers of a neural network into a single kernel, reducing memory access overhead and improving execution speed.
  • Precision Calibration: TensorRT supports reduced precision modes, such as mixed precision (FP16) and integer quantization (INT8). By reducing the number of bits used to represent numbers—often with minimal accuracy loss—developers can significantly accelerate math operations and reduce memory usage. This is a form of model quantization.
  • Kernel Auto-Tuning: The software automatically selects the best data layers and algorithms for the specific GPU architecture being used, ensuring maximum utilization of the hardware's parallel processing capabilities via CUDA.

Gerçek Dünya Uygulamaları

TensorRT , büyük miktarda veriyi minimum gecikmeyle işleme yeteneği sayesinde, zamanlamanın kritik olduğu bilgisayar görme ve karmaşık AI görevlerine dayanan sektörlerde yaygın olarak TensorRT . .

  1. Otonom Sistemler: Otomotiv sektöründe yapay zeka alanında, sürücüsüz arabalar detect , işaretleri ve engelleri anında detect için birden fazla kameradan gelen video akışlarını işlemek zorundadır. TensorRT kullanılarak, nesne algılama ağları gibi algılama modelleri kareleri milisaniyeler içinde analiz edebilir ve böylece aracın kontrol sistemi gecikme olmadan güvenlik açısından kritik kararlar alabilir.
  2. Endüstriyel Otomasyon: Modern fabrikalar, otomatik optik denetim için üretimde yapay zeka kullanıyor. Yüksek hızlı kameralar, montaj hatlarındaki ürünlerin görüntülerini yakalıyor ve TensorRT modeller, kusurları veya anormallikleri gerçek zamanlı olarak tespit ediyor. Bu, kalite kontrolünün yüksek hızlı üretim ortamlarına ayak uydurmasını sağlıyor ve genellikle NVIDIA platformu gibi uç yapay zeka cihazları doğrudan fabrika zemininde

Ultralytics YOLO TensorRT kullanma

Modern yapay zeka araçlarıyla TensorRT 'yi iş akışınıza entegre etmek çok kolay. TensorRT ultralytics package provides a seamless method to convert standard PyTorch models into TensorRT engines. This allows users to leverage the state-of-the-art architecture of Ultralytics YOLO26 with the hardware acceleration of NVIDIA GPUs. For teams looking to manage their datasets and training pipelines before export, the Ultralytics Platformu offers a comprehensive environment to prepare models for such high-performance deployment.

Aşağıdaki örnek, bir YOLO26 modelini bir TensorRT dosyasına nasıl dışa aktaracağınızı gösterir (.engine) ve bunu gerçek zamanlı çıkarım:

from ultralytics import YOLO

# Load the latest stable YOLO26 model (nano size)
model = YOLO("yolo26n.pt")

# Export the model to TensorRT format (creates 'yolo26n.engine')
# This step optimizes the computational graph for your specific GPU
model.export(format="engine")

# Load the optimized TensorRT engine for high-speed inference
trt_model = YOLO("yolo26n.engine")

# Run inference on an image source
results = trt_model("https://ultralytics.com/images/bus.jpg")

TensorRT vs. ONNX vs. Training Frameworks

It is important to distinguish TensorRT from other terms often heard in the model deployment landscape:

  • Vs. PyTorch/TensorFlow: Frameworks like PyTorch are primarily designed for model training and research, offering flexibility and ease of debugging. TensorRT is an inference engine designed solely for executing trained models as fast as possible. It is not used for training.
  • Vs. ONNX: The ONNX (Open Neural Network Exchange) format acts as an intermediary bridge between frameworks. While ONNX provides interoperability (e.g., moving a model from PyTorch to another platform), TensorRT focuses on hardware-specific optimization. Often, a model is converted to ONNX first, and then parsed by TensorRT to generate the final engine.

For developers aiming to maximize the performance of their AI agents or vision systems, understanding the transition from a training framework to an optimized runtime like TensorRT is a key step in professional MLOps.

Ultralytics topluluğuna katılın

Yapay zekanın geleceğine katılın. Küresel yenilikçilerle bağlantı kurun, işbirliği yapın ve birlikte büyüyün

Şimdi katılın