首页/AI/大模型部署与运维速查/TensorRT-LLM 与引擎编译部署
部署形态与选型43 条

TensorRT-LLM 与引擎编译部署

NVIDIA 官方的 LLM 推理优化库:把模型编译为针对特定 GPU 架构的 TensorRT 引擎再运行,风格接近「编译部署」;引擎与架构绑定,换卡需要重编。

典型流程(概念示例)

典型流程(概念示例)
# 1)获取官方容器环境
docker run --gpus all -it --rm nvcr.io/nvidia/tensorrt-llm/release
# 2)转换权重(按官方脚本)
python convert_checkpoint.py --model_dir <HF 模型目录> --dtype bfloat16 --output_dir ./ckpt
# 3)编译引擎
trtllm-build --checkpoint_dir ./ckpt --output_dir ./engine
# 4)启动服务(新版 CLI)
trtllm-serve <模型或引擎目录>
编译产物与 GPU 架构强绑定(Hopper 上编译的引擎不能直接在 Ada 上运行);升级驱动或框架后建议重编并回归。
工具链迭代快,命令以官方文档当前版本为准;上面演示的是「转换→编译→服务」这一固定结构。

易错点

  • 拿生产卡编译的引擎做本地验证常因架构不匹配失败,测试机与生产卡尽量同架构
  • 编译耗时长,模型与引擎目录要纳入版本管理,避免每次重建

数据来源与参考

  • NVIDIA TensorRT-LLM 官方文档与仓库(nvidia.github.io/TensorRT-LLM)
​‌‌​​‌​​​‌‌​‌‌‌‌​‌‌​​​‌‌​‌‌‌​​‌‌​​‌​‌‌‌​​‌‌‌​​​‌​‌‌​‌​​‌​‌‌‌​​​‌​‌‌​‌​​‌​‌‌​​​‌‌​‌‌‌​‌​​​‌‌​‌‌‌‌​​‌​‌‌‌​​‌‌​​​‌‌​‌‌​‌‌‌‌​‌‌​‌‌​‌