基于k8s云原生的模型推理架構(gòu)

Kubeflow是運行在K8S之上的一套技術(shù)棧,通過各種組件實現(xiàn)機器學(xué)習(xí)模型的訓(xùn)練和推理部署到云原生的模型平臺。

Kubeflow組件

Kubeflow提供了一大堆組件,涵蓋了機器學(xué)習(xí)的方方面面,為了對Kubeflow有個更直觀深入的了解,先整體看一下Kubeflow都有哪些組件,并對Kubeflow的主要組件進行簡單的介紹:

  • Central Dashboard:Kubeflow的dashboard看板頁面
  • Metadata:用于跟蹤各數(shù)據(jù)集、作業(yè)與模型
  • Jupyter Notebooks:一個交互式業(yè)務(wù)IDE編碼環(huán)境
  • Frameworks for Training:支持的ML框架,支持PyTorch,TensorFlow等
  • Pipelines:一個ML的工作流組件,用于定義復(fù)雜的ML工作流
  • Multi-Tenancy in Kubeflow:Kubeflow中的多租戶
  • Fairing:一個將code打包構(gòu)建image的組件
    Kubeflow中大多數(shù)組件的實現(xiàn)都是通過定義CRD來工作。目前Kubeflow主要的組件有:
  • Operator是針對不同的機器學(xué)習(xí)框架提供資源調(diào)度和分布式訓(xùn)練的能力(TF-Operator,PyTorch-Operator,Caffe2-Operator,MPI-Operator,MXNet-Operator);
  • Pipelines是一個基于Argo實現(xiàn)了面向機器學(xué)習(xí)場景的流水線項目,提供機器學(xué)習(xí)流程的創(chuàng)建、編排調(diào)度和管理,還提供了一個Web UI。
  • Katib Hyperparameter Tuning是基于各個Operator實現(xiàn)的超參數(shù)搜索和簡單的模型結(jié)構(gòu)搜索的系統(tǒng),支持并行搜索和分布式訓(xùn)練等。超參優(yōu)化在實際的工作中還沒有被大規(guī)模的應(yīng)用,所以這部分的技術(shù)還需要一些時間來成熟;
  • Serving支持部署各個框架訓(xùn)練好的模型的服務(wù)化部署和推理。Kubeflow提供基于TFServing,KFServing,Seldon等好幾種方案。
    • Seldon Core Serving
    • TensorFlow Serving(TFJob):提供對Tensorflow模型的在線部署,支持版本控制及無需停止線上服務(wù)、切換模型等
    • NVIDIA Triton Inference Server(Triton以前叫TensorRT),大語言模型版本是TensorRT-LLM,是LLM推理加速神器
    • TensorFlow Batch Prediction

TensorRT-LLM

TensorRT-LLM支持模型架構(gòu)定義、預(yù)訓(xùn)練權(quán)重編譯、推理加速,GPU 上的高效推理,做了 SOTA 級別的優(yōu)化,包含了一個可與 Triton Inference Server 集成的 backend,自帶主流的預(yù)定義熱門大語言模型,包括 baichuan、LlaMA、ChatGLM、BLOOM、GPT等。

LightLLM

LightLLM是商湯發(fā)布的推理服務(wù)框架,簡單高效,易于二次開發(fā)和其他框架的集成。

Knative

Knative 是谷歌發(fā)起的基于kubernetes平臺的Serverless 開源項目,致力將Serverless標(biāo)準(zhǔn)化。Kubernetes作為基礎(chǔ)設(shè)施,解決應(yīng)用編排和運行環(huán)境。Knative 將kubernetes和istio的復(fù)雜度進行抽象和隔離,解決了繁瑣的構(gòu)建,部署,服務(wù)治理步驟,并且基于開放標(biāo)準(zhǔn)使得服務(wù)變得可移植。

  • Isito作為通信基礎(chǔ)設(shè)施層,保證服務(wù)的運行可檢測、可配置、可追蹤;
  • Knative使用應(yīng)用模板和統(tǒng)一的運行環(huán)境來標(biāo)準(zhǔn)化服務(wù)的構(gòu)建、部署和管理;
  • Knative構(gòu)建在Kubernetes、Istio、Container的基礎(chǔ)上,以K8S的CRD形式存在。

Knative的三個組件(Serving、Build、Eventing)遵循了三個云原生最佳實踐的設(shè)計實現(xiàn)。

  • 服務(wù)的編排要實現(xiàn)計算資源彈性化
  • 服務(wù)的構(gòu)建和部署要實現(xiàn)高度自動化
  • 事件驅(qū)動基礎(chǔ)設(shè)施標(biāo)準(zhǔn)化
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,412評論 6 532
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,514評論 3 416
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事?!?“怎么了?”我有些...
    開封第一講書人閱讀 176,373評論 0 374
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,975評論 1 312
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 71,743評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,199評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,262評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,414評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,951評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,780評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,983評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,527評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,218評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,649評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,889評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,673評論 3 391
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 47,967評論 2 374

推薦閱讀更多精彩內(nèi)容