尧图网站设计 尧图网站设计YAOTU DESIGN
ARTICLE DETAIL

资讯详情

深耕网站设计与一线实操的经验洞察。

NVIDIA Triton Inference Server 架构解析与核心特性全景指南

NVIDIA Triton Inference Server 架构解析与核心特性全景指南 NVIDIA Triton Inference Server 架构解析与核心特性全景指南【免费下载链接】serverThe Triton Inference Server provides an optimized cloud and edge inferencing solution.项目地址: https://gitcode.com/gh_mirrors/server117/serverTriton Inference Server 是 NVIDIA 开源的 AI 推理服务软件用于将训练好的模型高效、规模化地部署为在线推理服务。本文以仓库 docs/introduction/index.md 为主干结合 架构文档、模型仓库文档 与 src 目录下的服务端源码系统讲解 Triton 的整体架构、请求处理链路、核心调度能力与可观测性设计帮助你理解一个模型仓库 一个服务进程如何支撑起生产级的多框架、多硬件推理平台。Triton Inference Server 是什么Triton Inference Server 是一个开源推理服务软件旨在简化 AI 推理的部署流程。它允许团队部署来自多种深度学习与机器学习框架的任意 AI 模型包括 TensorRT、PyTorch、ONNX、OpenVINO、Python、RAPIDS FIL 等。在硬件覆盖上Triton 支持在 NVIDIA GPU、x86 与 ARM CPU、以及 AWS Inferentia 上运行可部署于云、数据中心、边缘设备与嵌入式设备等多种环境。Triton 针对多种查询类型提供优化的性能包括实时推理、批量推理、模型集成ensemble以及音频/视频流式推理。Triton Inference Server 同时也是 NVIDIA AI Enterprise 软件平台的一部分该平台用于加速数据科学流水线并简化生产级 AI 的开发与部署。从仓库版本看当前 TRITON_VERSION 标识为2.74.0devmain 分支跟踪下一版本的开发进度最新发布版本为 2.72.0。仓库根目录的 README.md 给出了完整的三步快速部署示例对应快速入门指南见 docs/getting_started/quickstart.md。Triton 高维架构一次推理请求的完整旅程Triton 的高层架构可以概括为一条清晰的请求处理链路架构图如下完整的链路如下见 docs/user_guide/architecture.md模型仓库Model Repository一个基于文件系统的模型仓库存放 Triton 将要对外提供推理服务的所有模型。模型仓库的布局规范详见 docs/user_guide/model_repository.md。请求入口推理请求通过 HTTP/REST 或 GRPC 协议 到达服务端或者通过 进程内 C API 直接注入。在源码层面HTTP 服务由 src/http_server.cc 实现GRPC 服务由 src/grpc 目录实现进程入口为 src/main.cc。逐模型调度器Per-model Scheduler请求被路由到对应模型的调度器。Triton 实现了多种 调度与批处理算法可针对每个模型独立配置。后端执行Backend每个模型的调度器可选择性地对推理请求进行批处理batching然后将请求交给与该模型类型对应的 backend。backend 使用批处理请求中提供的输入执行推理产生请求的输出并返回给客户端。Triton 还提供 backend C API 目录与 qa 目录中大量以L0_backend_*命名的测试如 qa/L0_backend_identity、qa/L0_backend_onnxruntime正是围绕不同 backend 的加载与推理行为进行验证。模型仓库Triton 服务的文件系统底座模型仓库是 Triton 推理服务的起点。仓库路径通过--model-repository启动参数指定且可以多次指定以加载多个仓库。其标准布局如下摘自 docs/user_guide/model_repository.md$ tritonserver --model-repositorymodel-repository-path对应的仓库目录布局必须为model-repository-path/ model-name/ [config.pbtxt] [output-labels-file ...] [configs]/ [custom-config-file ...] version/ model-definition-file version/ model-definition-file ... model-name/ ...要点说明顶层目录下包含零个或多个model-name子目录每个子目录存放对应模型的所有仓库信息config.pbtxt描述该模型的 模型配置。部分模型必须提供该文件另一些模型可省略由 Triton 自动生成配置每个模型目录必须包含至少一个数字命名的版本子目录用于表示模型的版本同一模型的多个版本可共存由 Triton 统一管理每个模型由特定的 backend 执行版本子目录内需放置该 backend 要求的模型文件例如 TensorRT、PyTorch、ONNX、OpenVINO 等框架后端所需的对应格式模型文件。仓库内的示例模型仓库位于 docs/examples/model_repository其中包含 densenet_onnx 等演示模型docs/examples/fetch_models.sh 脚本用于从公共模型库拉取缺失的模型定义文件。另外模型仓库也可放在 Google Cloud Storage、Amazon S3、Azure Storage 等云存储中仓库根目录的 qa/L0_storage_S3、qa/L0_storage_azure、qa/L0_storage_swiftstack 等测试验证了这些存储后端的可用性。调度与批处理模型执行的交通管制Triton 为每个模型独立配置调度与批处理算法主要区分为**无状态模型stateless与有状态模型stateful**两类详见 docs/user_guide/architecture.md无状态模型推理请求之间不维护状态每次推理相互独立如图像分类、目标检测等 CNN 模型。可使用 默认调度器 或 动态批处理器。有状态模型请求之间需要维护状态一系列推理请求构成一个序列sequence必须被路由到同一个模型实例以保证状态正确更新且模型可能需要 Triton 提供开始/结束等控制信号。此类模型必须使用 序列批处理器它保证同一序列的所有请求命中同一模型实例并通过 correlation ID 与模型通信。在并发执行方面见 docs/user_guide/model_execution.md默认情况下同一模型的多个并发请求会在 GPU 上串行执行通过模型配置中的instance-group字段可以为一个模型配置多个并行执行实例instance例如配置 3 个实例后前 3 个请求可立即并行执行第 4 个请求需等待其中之一完成不同模型的请求可同时被调度到 GPU 上并行执行GPU 的硬件调度器负责并行计算CPU 上的模型执行则由操作系统线程调度。这一机制保证了 Triton 在单机多模型、多实例场景下可以最大化硬件利用率相关行为在 qa/L0_batcher、qa/L0_dyna_sequence_batcher、qa/L0_sequence_batcher 等测试中得到充分验证。Triton 的核心特性清单根据 docs/introduction/index.md 与仓库根 README.mdTriton 的主要特性包括支持多种深度学习框架TensorRT、PyTorch、ONNX、OpenVINO 等支持多种机器学习框架如 RAPIDS FIL随机森林/梯度提升树等模型并发模型执行多模型、多实例并行执行见 docs/user_guide/model_execution.md动态批处理Dynamic Batching在运行时将多个请求动态合并成批提升吞吐见 docs/user_guide/batcher.md序列批处理与隐式状态管理面向有状态模型支持序列路由与状态自动管理见 docs/user_guide/implicit_state_management.mdBackend API允许添加自定义 backend 以及预处理/后处理操作还支持用 Python 编写自定义 backendPython-based backends仓库 qa/L0_backend_python 下大量测试用例覆盖了 Python backend 的生命周期、BLS、decoupled 等场景模型流水线通过 Ensembling模型集成 或 Business Logic ScriptingBLS 将多个模型编排为复杂推理流水线集成模型的配置样例可参考 qa/ensemble_models 目录HTTP/REST 与 GRPC 推理协议基于社区开发的 KServe 协议 目录如 extension_generate.md、extension_model_repository.md 等进程内 C API 与 Java API允许 Triton 直接链接进应用程序适用于边缘设备等进程内in-process使用场景说明见 docs/customization_guide/inprocess_c_api.md 与 docs/customization_guide/inprocess_java_api.md指标Metrics提供 GPU 利用率、服务吞吐、服务延迟等指标见 docs/user_guide/metrics.md。推理协议与模型管理 API客户端可通过两种网络协议与 Triton 通信详见 docs/customization_guide/inference_protocols.mdHTTP/REST 与 GRPC均基于 KServe 社区提出的标准推理协议Triton 同时实现了对 KServe 协议的扩展以启用全部能力。GRPC 还提供推理 RPC 的双向流式版本允许在一条 GRPC 流上发送序列化的推理请求/响应。官方通常推荐使用一元unary版本流式版本仅在特定场景下使用例如需要将序列请求路由到负载均衡后面的同一 Triton 实例、或需要保持网络上的请求/响应顺序时。进程内 C API直接链接进应用绕过网络开销。对正在服务的模型Triton 提供专门的模型管理 API可通过 HTTP/REST、GRPC 或 C API 访问见 docs/user_guide/model_management.md。Triton 支持三种模型控制模式NONE默认启动时尝试加载仓库中所有模型运行期间忽略仓库变更模型加载/卸载请求返回错误EXPLICIT启动时仅加载通过--load-model明确指定的模型使用--load-model*可加载全部模型注意*必须是唯一的--load-model参数运行期间所有加载/卸载动作必须通过模型控制协议显式发起POLL启动时加载所有模型并周期性轮询模型仓库目录自动发现模型文件的增删改。健康检查与指标为 Kubernetes 等部署框架而生Triton 提供就绪readiness与存活liveness健康检查端点以及利用率、吞吐、延迟等指标便于将 Triton 集成进 Kubernetes 等部署框架。仓库 deploy 目录提供了 AWS、GCP、OCI、k8s-onprem、fleetcommand 等多套 Helm Chart 部署模板正是这些能力在生产编排环境中的落地示例。指标方面见 docs/user_guide/metrics.mdTriton 默认在http://localhost:8002/metrics暴露 Prometheus 格式指标可直接curl查看可通过--allow-metricsfalse关闭全部指标用--allow-gpu-metricsfalse/--allow-cpu-metricsfalse分别关闭 GPU/CPU 指标用--metrics-port修改端口用--metrics-address单独指定指标端点地址HTTP 服务启用时默认复用--http-address用--metrics-interval-ms调整指标的轮询/更新间隔Per Request 类指标不受该间隔影响。在源码层面健康检查、指标与推理服务的端点统一由 src/http_server.cc 中的 HTTPService / Metrics Service 提供其启动日志Started HTTPService、Started Metrics Service与 docs/getting_started/quickstart.md 中的运行示例一致。实战三步启动 Triton 并完成一次推理结合 README.md 与 docs/getting_started/quickstart.md可以用三步快速跑通 Triton 的完整推理流程Step 1创建示例模型仓库git clone -b r26.08 https://github.com/triton-inference-server/server.git cd server/docs/examples ./fetch_models.shStep 2启动 Triton使用 NGC 官方容器启动--gpus1表示将 1 块 GPU 提供给 Triton需预先安装 NVIDIA Container Toolkitdocker run --gpus1 --rm --nethost -v ${PWD}/model_repository:/models nvcr.io/nvidia/tritonserver:26.08-py3 tritonserver --model-repository/models --model-control-mode explicit --load-model densenet_onnx启动后控制台会打印模型加载状态表与各服务端点日志--------------------------------------- | Model | Version | Status | --------------------------------------- | model_name | v | READY | --------------------------------------- ... I1002 21:58:57.891440 62 grpc_server.cc:3914] Started GRPCInferenceService at 0.0.0.0:8001 I1002 21:58:57.893177 62 http_server.cc:2717] Started HTTPService at 0.0.0.0:8000 I1002 21:58:57.935518 62 http_server.cc:2736] Started Metrics Service at 0.0.0.0:8002所有模型显示READY表示加载成功若加载失败状态列会报告失败原因。无 GPU 的 CPU-only 系统只需去掉--gpus参数其余命令一致但无法加载需要 GPU 的模型配置。Step 3发送推理请求用curl访问就绪端点确认服务正常返回 200 即就绪非 200 表示未就绪curl -v localhost:8000/v2/health/ready然后使用 SDK 容器中的image_client示例程序对 densenet_onnx 模型发起图像分类请求取置信度最高的 3 个类别docker run -it --rm --nethost nvcr.io/nvidia/tritonserver:26.08-py3-sdk /workspace/install/bin/image_client -m densenet_onnx -c 3 -s INCEPTION /workspace/images/mug.jpg预期输出Request 0, batch size 1 Image /workspace/images/mug.jpg: 15.346230 (504) COFFEE MUG 13.224326 (968) CUP 10.422965 (505) COFFEEPOT从源码视角看请求处理链路在仓库源码中Triton 的请求处理链路有清晰的落点进程入口src/main.cc 负责解析--model-repository、--model-control-mode、--load-model等启动参数初始化各服务组件HTTP/REST 服务src/http_server.cc 与 src/http_server.h 实现 KServe V2 推理协议与健康检查、指标端点GRPC 服务src/grpc 目录下的源码实现GRPCInferenceService及双向流式推理 RPC共享内存与数据通路src/shared_memory_manager.cc 管理 CPU/GPU 共享内存支持零拷贝输入输出命令行解析src/command_line_parser.cc 集中处理全部 CLI 选项所有参数均可通过tritonserver --help查看。对应地qa 目录提供了覆盖 HTTP、GRPC、指标、模型管理、调度器等各层面的端到端测试如 qa/L0_http/http_test.py、qa/L0_grpc、qa/L0_metrics、qa/L0_model_management 等是理解各组件行为与验证部署正确性的第一手参考。总结Triton Inference Server 的核心设计可以概括为一处模型仓库、多种协议入口、逐模型可配调度、后端可扩展、全链路可观测它以文件系统模型仓库作为统一输入通过 KServe 标准协议对外提供 HTTP/REST 与 GRPC 服务并支持进程内 C/Java API针对无状态与有状态模型分别提供动态批处理与序列批处理通过 backend 机制接入 TensorRT、PyTorch、ONNX、OpenVINO、Python 等框架支持集成与 BLS 流水线编排同时以健康检查端点与 Prometheus 指标无缝融入 Kubernetes 等生产编排环境。无论部署在 NVIDIA GPU、x86/ARM CPU 还是 AWS Inferentia 上这套架构都能在云、数据中心、边缘与嵌入式场景下提供一致而高效的推理服务能力。【免费下载链接】serverThe Triton Inference Server provides an optimized cloud and edge inferencing solution.项目地址: https://gitcode.com/gh_mirrors/server117/server创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表