SuperAPI AI 工具集

TensorRT-LLM

TensorRT-LLM 是 NVIDIA 开源的大语言模型推理优化库,基于 PyTorch 构建,提供易用的 Python LLM API,通过在 NVIDIA GPU 上应用自定义内核、运行时算法优化等手段,高效执行 LLM 与视觉生成模型的推理。

访问官网

工具介绍

TensorRT-LLM 是 NVIDIA 推出的开源库,用于优化大语言模型(LLM)和视觉生成(Visual Gen)的推理。它提供面向 NVIDIA GPU 的先进优化能力,帮助用户在推理场景中获得更好的性能表现。

主要能力

  • 针对常见推理操作的自定义内核,覆盖 attention、GEMM、MoE 等。
  • 运行时算法优化,包括 Prefill-Decode 分离(disaggregation)、Wide Expert Parallelism、Speculative Decoding 等。
  • 高层 Python LLM API,支持从单 GPU 到多 GPU、多节点部署的多种推理设置。
  • 内置多种并行策略与高级特性支持。
  • 与 NVIDIA Dynamo、Triton Inference Server 等推理生态无缝集成。
  • 模块化、易于修改:PyTorch 原生架构允许开发者实验运行时或扩展功能;多个流行模型已预定义,并可用原生 PyTorch 代码自定义。

使用方式 官方文档提供以下入口:

  • Quick Start Guide(快速开始指南)
  • Running DeepSeek(运行 DeepSeek)
  • Installation Guide(安装指南)
  • Supported Hardware, Models, and other Software(支持的硬件、模型及其他软件)
  • Benchmarking Performance(性能基准测试)

TensorRT-LLM 提供 Python 与 C++ 运行时组件,用于以高性能方式编排推理执行。

适用场景

  • 在 NVIDIA GPU 上部署 LLM 推理,从单 GPU 到多 GPU、多节点场景。
  • 需要自定义内核、量化、投机解码、长上下文优化、MoE 优化等高级推理优化的场景。
  • 与 Triton Inference Server、NVIDIA Dynamo 等推理服务生态配合使用。
  • 视觉生成(扩散模型)推理。

最新动态(官方列出的部分更新)

  • KV Cache Compression in TensorRT LLM
  • GVR V2:Self-Sampling and Multi-Thresholding for Faster Exact Top-K
  • Accelerating Video Generation with GEMM Quantization, Attention Quantization and Skip Softmax Attention
  • Evaluating Agentic Serving with Trace Replay and Job-Level Metrics
  • DeepSeek-V4 on NVIDIA Blackwell:Model-Specific and Agentic-Workload Optimizations
  • Scaling Video Generation Across NVL72 Rack
  • Joint Optimization of Agent Applications and TensorRT-LLM
  • Tuning CUDA Graph Batch Sizes for Higher Output Throughput
  • DWDP:Distributed Weight Data Parallelism for High-Performance LLM Inference on NVL72
  • Optimizing MoE Communication with One-Sided AlltoAll Over NVLink
  • Sparse Attention in TensorRT LLM
  • Accelerating Long-Context Inference with Skip Softmax Attention
  • Optimizing DeepSeek-V3.2 on NVIDIA Blackwell GPUs
  • Inference Time Compute Implementation
  • Combining Guided Decoding and Speculative Decoding
  • ADP Balance Strategy
  • Running a High-Performance GPT-OSS-120B Inference Server
  • Scaling Expert Parallelism(Part 1 与 Part 2)
  • N-Gram Speculative Decoding
  • Disaggregated Serving
  • 支持扩散模型用于视觉生成
  • 对 OpenAI GPT-OSS-120B、GPT-OSS-20B 的 Day-0 支持
  • 对 LG AI Research EXAONE 4.0 的 Day-0 支持
  • Blackwell 上 Meta Llama 4 Maverick 突破 1,000 TPS/User
  • Llama 4 在 B200 GPU 上超过 40,000 tokens/秒
  • 完全开源,开发移至 GitHub
  • 官方发布 DeepSeek R1 性能基准测试最佳实践
  • NVIDIA Blackwell 以 TensorRT LLM 实现 DeepSeek-R1 世界纪录推理性能
  • 更早的更新还包括 KV Cache Reuse 优化、Bing 搜索优化、Llama 3.3 70B 推理吞吐提升、Speculative Decoding 支持、Jetson AGX Orin 支持、Multiblock Attention、多语言优化、SD3 INT8 量化、ComfyUI 部署等。

弃用政策 从 1.0 版本开始,TensorRT LLM 采用弃用政策:弃用通知记录在 Release Notes 中;被弃用的 API、方法、类或参数会在源码中标明弃用时间,使用时产生运行时弃用警告。弃用后提供 3 个月迁移期,期间被弃用项仍可用但会告警;迁移期结束后按语义化版本规则移除(主版本变更可能包含破坏性移除)。弃用范围包括整体 API/方法/类弃用,以及仅部分参数弃用(方法本身仍可用,被弃用参数将在未来版本移除)。

遥测数据收集 TensorRT-LLM 默认收集匿名遥测数据,用于汇总了解使用模式并确定工程优先级,数据无法追溯到任何个人用户。不收集提示词、输出、模型权重、模型路径、tokenizer 路径、可识别用户信息、原始自由格式配置字符串或持久标识符;部署标识符为临时的、每次部署随机生成,且不与用户关联。收集的数据包括:入口点(如 LLM API、CLI、serve 命令)、部署时长(通过周期性心跳)、GPU SKU、数量、内存与 CUDA 版本、公开模型架构标识符(可能为上游 checkpoint 架构或公开文档化的 TensorRT-LLM 运行时架构;白名单之外的标识符仅发送确定性 SHA-256 哈希)、并行配置(TP/PP/CP/MoE-EP/MoE-TP 大小)、量化算法、dtype、KV cache dtype、系统信息(OS 平台、Python 版本、CPU 架构、CPU 数量)、TRT-LLM 版本与后端、特性汇总标志(LoRA、speculative decoding、prefix caching、CUDA graphs、chunked context、data parallelism)、分离式服务元数据(角色与部署 ID)、进程本地 LLM 生命周期信息。

限制与说明 资料未提供具体收费、价格、平台支持细节、性能数字的完整测评或具体使用步骤,相关内容暂未提供。