前面文章介紹微軟 Teams 會議系統(tǒng)、微信軟件與騰訊 PCG 服務(wù)三個 Triton 推理服務(wù)器的成功案例，讓大家對 Triton 有初步的認(rèn)知，但別誤以為這個軟件只適合在大型的服務(wù)類應(yīng)用中使用，事實上 Triton 能適用于更廣泛的推理環(huán)節(jié)中，并且在越復(fù)雜的應(yīng)用環(huán)境中就越能展現(xiàn)其執(zhí)行成效。

在說明 Triton 推理服務(wù)器的架構(gòu)與功能之前，我們需要先了解一個推理服務(wù)器所需要面對并解決的問題。

與大部分的服務(wù)器軟件所需要的基本功能類似，一個推理服務(wù)器也得接受來自不同用戶端所提出的各種要求（request）然后做出回應(yīng)（response），并且對系統(tǒng)的處理進(jìn)行性能優(yōu)化與穩(wěn)定性管理。

但是推理計算需要面對深度學(xué)習(xí)領(lǐng)域的各式各樣推理模型，包括圖像分類、物件檢測、語義分析、語音識別等不同應(yīng)用類別，每種類別還有不同神經(jīng)網(wǎng)絡(luò)算法與不同框架所訓(xùn)練出來的模型格式等。此外，我們不能對任務(wù)進(jìn)行單純的串行隊列（queue）方式處理，這會使得任務(wù)等待時間拖得很長，影響使用的體驗感，因此必須對任務(wù)進(jìn)行并行化處理，這里就存在非常復(fù)雜的任務(wù)管理技巧。

下面列出一個推理服務(wù)器所需要面對的技術(shù)問題：

1.支持多種模型格式：至少需要支持普及度最高的

2.TensorFlow 的 GraphDef 與 SavedMode 中一種以上格式

(1) PyTorch 的 TorchScript 格式

(2) ONNX 開放標(biāo)準(zhǔn)格式

(3) 其他：包括自定義模型格式

3.支持多種查詢類型，包括

(1) 在線的實時查詢：盡量降低查詢的延遲（latency）時間

(2) 離線的批量處理：盡量提高查詢的通量（throughput）

(3) 流水線傳輸?shù)淖R別號管理等工作

4.支持多種部署方式：包括

(1) 企業(yè)的 GPU 或 CPU 計算設(shè)備

(2) 公共云或數(shù)據(jù)中心

5.對模型進(jìn)行最佳縮放處理：讓個別模型提供更好的性能

6.優(yōu)化多個 KPI：包括

(1) 硬件利用率

(2) 模型推理識別時間

(3) 總體成本（TCO）

7.提高系統(tǒng)穩(wěn)定性：需監(jiān)控模型狀態(tài)并解決問題以防止停機(jī)

在了解推理服務(wù)器所需要解決的關(guān)鍵問題之后，接著來看看下方的 Triton 系統(tǒng)高階架構(gòu)圖，就能更清楚每個板塊所負(fù)責(zé)的任務(wù)與使用的對應(yīng)技術(shù)。

Triton 推理服務(wù)器采用屬于 “主從（client-server）” 架構(gòu)的系統(tǒng)，由圖中的四個板塊所組成：

1.模型倉（Model Repostory）：存放 Triton 服務(wù)器所要使用的模型文件與配置文件的存儲設(shè)備，可以是本地服務(wù)器的文件系統(tǒng)，也可以使用 Google、AWS、Azure 等云存儲空間，只要遵循 Triton 服務(wù)器所要求的規(guī)范就可以；

2.客戶端應(yīng)用（Client Application）：基于 Triton 用戶端 Python / C++ / Java 庫所撰寫，可以在各種操作系統(tǒng)與 CPU 架構(gòu)上操作，對 Triton 服務(wù)器提交任務(wù)請求，并且接受返回的計算結(jié)果。這是整個 Triton 推理應(yīng)用中代碼量最多的一部分，也是開發(fā)人員需要花費最多心思的部分，在后面會有專文講解。

3.HTTP / gPRC 通訊協(xié)議：作為用戶端與服務(wù)端互動的通訊協(xié)議，開發(fā)人員可以根據(jù)實際狀況選擇其中一種通訊協(xié)議進(jìn)行操作，能透過互聯(lián)網(wǎng)對服務(wù)器提出推理請求并返回推理結(jié)果，如下圖所示：

使用這類通訊協(xié)議有以下優(yōu)點：

(1)支持實時、批處理和流式推理查詢，以獲得最佳應(yīng)用程序體驗

(2)提供高吞吐量推理，同時使用動態(tài)批處理和并發(fā)模型執(zhí)行來滿足緊張的延遲預(yù)算

(3)模型可以在現(xiàn)場制作中更新，而不會中斷應(yīng)用程序

4.推理服務(wù)器（Inference Server）：這是整個 Triton 服務(wù)器最核心且最復(fù)雜的部分，特別在 “性能”、“穩(wěn)定”、“擴(kuò)充” 這三大要求之間取得平衡的管理，主要包括以下幾大功能板塊：

(1) C 開發(fā)接口：

在服務(wù)器內(nèi)的代碼屬于系統(tǒng)底層機(jī)制，主要由 NVIDIA 系統(tǒng)工程師進(jìn)行維護(hù)，因此只提供性能較好的 C 開發(fā)接口，一般應(yīng)用工程師可以忽略這部分，除非您有心深入 Triton 系統(tǒng)底層進(jìn)行改寫。

(2) 模型管理器（Model Management）：

支持多框架的文件格式并提供自定義的擴(kuò)充能力，目前已支持 TensorFlow 的 GraphDef 與 SavedModel 格式、ONNX、PyTorch TorchScript、TensorRT、用于基于樹的 RAPIDS FIL 模型、OpenVINO 等模型文件格式，還能使用自定義的 Python / C++ 模型格式；

(3) 模型的推理隊列調(diào)度器（Per-Model Scheduler Queues）：

將推理模型用管道形式進(jìn)行管理，將一個或多個模型的預(yù)處理或后處理進(jìn)行邏輯排列，并管理模型之間的輸入和輸出張量的連接，任何的推理請求都會觸發(fā)這個模型管道。這部分還包含以下兩個重點：

并發(fā)模型執(zhí)行（Concurrent Model Execution）：允許同一模型的多個模型和 / 或多個實例在同一系統(tǒng)上并行執(zhí)行，系統(tǒng)可能有零個、一個或多個 GPU。

模型和調(diào)度程序（Models And Schedulers）：支持多種調(diào)度和批量處理算法，可為每個模型單獨選擇無狀態(tài)（stateless）、有狀態(tài)（stateful）或集成（ensemble）模式。對于給定的模型，調(diào)度器的選擇和配置是通過模型的配置文件完成的。

(4) 計算資源的優(yōu)化處理：

這是作為服務(wù)器軟件的最重要工作之一，就是要將設(shè)備的計算資源充分調(diào)度，并且優(yōu)化總體計算性能，主要使用以下三種技術(shù)。

支持異構(gòu)計算模式：可部署在純 x86 與 ARM CPU 的計算設(shè)備上，也支持裝載 NVIDIA GPU 的計算設(shè)備。

動態(tài)批量處理（Dynamic batching）技術(shù)：對支持批處理的模型提供多個內(nèi)置的調(diào)度和批處理算法，并結(jié)合各個推理請求以提高推理吞吐量，這些調(diào)度和批量處理決策對請求推理的客戶端是透明的。

批量處理推理請求分為客戶端批量處理和服務(wù)器批量處理兩種，通過將單個推理請求組合在一起來實現(xiàn)服務(wù)器批處理，以提高推理吞吐量；

構(gòu)建一個批量處理緩存區(qū)，當(dāng)達(dá)到配置的延遲閾值后便啟動處理機(jī)制；

調(diào)度和批處理決策對請求推斷的客戶機(jī)是透明的，并且根據(jù)模型進(jìn)行配置。

c.并發(fā)模型（Concurrent model）運行：多個模型或同一模型的多個實例，可以同時在一個 GPU 或多個 GPU 上運行，以滿足不同的模型管理需求。

(5) 框架后端管理器（Framework Backends）：

Triton 的后端就是執(zhí)行模型的封裝代碼，每種支持的框架都有一個對應(yīng)的后端作為支持，例如 tensorrt_backend 就是支持 TensorRT 模型推理所封裝的后端、openvino_backend 就是支持 openvine 模型推理所封裝的后端，目前在 Triton 開源項目里已經(jīng)提供大約 15 種后端，技術(shù)人員可以根據(jù)開發(fā)無限擴(kuò)充。

要添加一個新的后臺是相當(dāng)復(fù)雜的過程，因此在本系列文章中并不探索，這里主要說明以下 Triton 服務(wù)器對各個后端的管理機(jī)制，主要是以下重點：

采用 KFServing 的新社區(qū)標(biāo)準(zhǔn) gRPC 和 HTTP/REST 數(shù)據(jù)平面（data plane）v2 協(xié)議（如下圖），這是 Kubernetes 上基于各種標(biāo)準(zhǔn)的無服務(wù)器推理架構(gòu)

通過配置自動化和自動擴(kuò)展簡化 Kubernetes 中的推理服務(wù)部署

透明地處理負(fù)載峰值，即使請求數(shù)量顯著增加，請求的服務(wù)也將繼續(xù)順利運行

可以通過定義轉(zhuǎn)換器，輕松地將標(biāo)記化和后處理等預(yù)處理步驟包含在部署中

可以用 NGC 的 Helm 命令在 Kubernetes 中部署 Triton，也可以部署為容器微服務(wù)，為 GPU 和 CPU 上的預(yù)處理或后處理和深度學(xué)習(xí)模型提供服務(wù)，也能輕松部署在數(shù)據(jù)中心或云平臺上

將推理實例進(jìn)行微服務(wù)處理，每個實例都可以在 Kubernetes 環(huán)境中獨立擴(kuò)展，以獲得最佳性能

通過這種新的集成，可以輕松地在 Kubernetes 使用 Triton 部署高性能推理

以上是 Triton 推理服務(wù)器的高級框架與主要特性的簡介，如果看完本文后仍感覺有許多不太理解的部分，這是正常的現(xiàn)象，因為整個 Triton 系統(tǒng)集成非常多最先進(jìn)的技術(shù)在內(nèi)，并非朝夕之間就能掌握的。

后面的內(nèi)容就要進(jìn)入 Triton 推理服務(wù)器的環(huán)境安裝與調(diào)試，以及一些基礎(chǔ)范例的執(zhí)行環(huán)節(jié)，透過這些實際的操作，逐步體驗 Triton 系統(tǒng)的強(qiáng)大。

審核編輯：湯梓紅