nvidia-docker2.0是一個簡單的包，它主要通過修改docker的配置文件“/etc/docker/daemon.json”來讓docker使用NVIDIA Container runtime。

本文操作環(huán)境：Windows10系統(tǒng)、Docker 20.10.11版、Dell G3電腦。

NVidia Docker介紹

NVIDIA于2016年開始設計NVIDIA-Docker已便于容器使用NVIDIA GPUs。 第一代nvidia-docker1.0實現(xiàn)了對docker client的封裝，并在容器啟動時，將必要的GPU device和libraries掛載到容器中。但是這種設計的方式高度的與docker運行時耦合，缺乏靈活性。存在的缺陷具體如下:

設計高度與docker耦合，不支持其它的容器運行時。如: LXC, CRI-O及未來可能會增加的容器運行時。

不能更好的利用docker生態(tài)的其它工具。如: docker compose。

不能將GPU作為調(diào)度系統(tǒng)的一種資源來進行靈活的調(diào)度。

完善容器運行時對GPU的支持。如: 自動的獲取用戶層面的NVIDIA Driver libraries, NVIDIA kernel modules, device ordering等。

基于上面描述的這些弊端，NVIDIA開始了對下一代容器運行時的設計: nvidia-docker2.0。

nvidia-docker 2.0 的實現(xiàn)機制

先簡單介紹下nvidia-docker 2.0, containerd，nvidia-container-runtime，libnvidia-container以及runc直接的關系。

它們之間的關系可以通過下面這張圖關聯(lián)起來:

nvidia-docker 2.0

nvidia-docker2.0 是一個簡單的包，它主要通過修改docker的配置文件/etc/docker/daemon.json來讓docker使用NVIDIA Container runtime。

nvidia-container-runtime

nvidia-container-runtime 才是真正的核心部分，它在原有的docker容器運行時runc的基礎上增加一個prestart hook，用于調(diào)用libnvidia-container庫。

libnvidia-container

libnvidia-container 提供一個庫和一個簡單的CLI工具，使用這個庫可以使NVIDIA GPU被Linux容器使用。

Containerd

Containerd主要負責的工作是:

• 管理容器的生命周期(從容器的創(chuàng)建到銷毀)

• 拉取/推送容器鏡像

• 存儲管理(管理鏡像及容器數(shù)據(jù)的存儲)

• 調(diào)用runc 運行容器

• 管理容器的網(wǎng)絡接口及網(wǎng)絡

當containerd接收到請求之后，做好相關的準備工作，可以選擇自己調(diào)用runc也可以通過創(chuàng)建containerd-shim再去調(diào)用runc，而runc基于OCI文件對容器進行創(chuàng)建。 上面就是普通容器創(chuàng)建的基本流程。

RunC

RunC 是一個輕量級的工具，它是用來運行容器的，只用來做這一件事，并且這一件事要做好。我們可以認為它就是個命令行小工具，可以不用通過 docker 引擎，直接運行容器。事實上，runC 是標準化的產(chǎn)物，它根據(jù) OCI 標準來創(chuàng)建和運行容器。而 OCI(Open Container Initiative)組織，旨在圍繞容器格式和運行時制定一個開放的工業(yè)化標準。

直接使用RunC的命令行即可以完成創(chuàng)建一個容器，并提供了簡單的交互能力。

上面已經(jīng)介紹個各個組件的作用以及它們之間的關系，接下來詳細的描述下這張圖:

正常創(chuàng)建一個容器的流程是這樣的:

docker --> dockerd --> containerd--> containerd-shim -->runc --> container-process

docker客戶端將創(chuàng)建容器的請求發(fā)送給dockerd, 當dockerd收到請求任務之后將請求發(fā)送給containerd, containerd經(jīng)過查看校驗啟動containerd-shim或者自己來啟動容器進程。

創(chuàng)建一個使用GPU的容器

創(chuàng)建GPU容器的流程如下:

docker--> dockerd --> containerd --> containerd-shim--> nvidia-container-runtime --> nvidia-container-runtime-hook --> libnvidia-container --> runc -- > container-process

基本流程和不使用GPU的容器差不多，只是把docker默認的運行時替換成了NVIDIA自家的nvidia-container-runtime。

這樣當nvidia-container-runtime創(chuàng)建容器時，先執(zhí)行nvidia-container-runtime-hook這個hook去檢查容器是否需要使用GPU(通過環(huán)境變NVIDIA_VISIBLE_DEVICES來判斷)。如果需要則調(diào)用libnvidia-container來暴露GPU給容器使用。否則走默認的runc邏輯。

說到這里nvidia-docker2.0的大體機制基本就通了。但是涉及到的nvidia-container-runtime, libnvidia-container, containerd,runc這些項目， 這本篇文章里面就不一一介紹了。如果感興趣可以自行去探索學習。這些項目的地址在文章中都已經(jīng)做個相關的鏈接。

推薦學習：《Docker視頻教程》