Kā izmantot NVIDIA GPU Docker konteineros Ubuntu 22.04 LTS

Ka Izmantot Nvidia Gpu Docker Konteineros Ubuntu 22 04 Lts



Varat nosūtīt savu NVIDIA GPU Docker konteineros un palaist CUDA programmas savā NVIDIA GPU no šiem Docker konteineriem. Šī ir ļoti noderīga funkcija AI (mākslīgā intelekta) apguvei. Iespēja palaist AI kodus (t.i., Tensorflow) Docker konteineros, ietaupīsiet daudz laika. Varat arī izmēģināt savus AI kodus dažādās CUDA versijās, nemainot CUDA versiju datorā.

Darbošanās ar NVIDIA draiveriem un CUDA versijām datorā var radīt problēmas, piemēram, NVIDIA draiveri nedarbosies vai jūs varat atstāt melnu/zilu nāves ekrānu. Tāpēc vienmēr ir ieteicams atstāt sistēmu tādu, kāda tā ir, un veikt izmaiņas izolētā vidē, piemēram, Docker konteineros. Tādējādi jūsu galvenais dators ir tīrs (no nevajadzīgiem izstrādes rīkiem).







Šajā rakstā es jums parādīšu, kā iestatīt Docker CE un NVIDIA docker Ubuntu 22.04 LTS, lai jūs varētu piekļūt sava datora NVIDIA GPU no Docker konteineriem un palaist CUDA programmas savā NVIDIA GPU.



Satura rādītājs:

  1. Pārbaude, vai Ubuntu 22.04 ir instalēti oficiālie NVIDIA GPU draiveri
  2. Docker CE atkarību instalēšana uz Ubuntu 22.04
  3. Docker CE GPG atslēgu instalēšana Ubuntu 22.04
  4. Docker CE repozitorija instalēšana Ubuntu 22.04
  5. Docker CE instalēšana Ubuntu 22.04
  6. Ubuntu 22.04 LTS pieteikšanās lietotāja pievienošana dokera grupai
  7. Pārbaude, vai Docker CE ir pareizi instalēts Ubuntu 22.04
  8. NVIDIA Container Toolkit GPG Keys instalēšana Ubuntu 22.04
  9. NVIDIA Container Toolkit repozitorija instalēšana Ubuntu 22.04
  10. Nvidia-docker draiveru instalēšana Ubuntu 22.04
  11. Pārbaude, vai NVIDIA GPU ir pieejams no Docker konteineriem Ubuntu 22.04
  12. Secinājums
  13. Atsauces

Pārbaude, vai Ubuntu 22.04 LTS ir instalēti oficiālie NVIDIA GPU draiveri:

Pirms darba sākšanas palaidiet šo komandu, lai pārbaudītu, vai datorā ir instalēts NVIDIA GPU.



$ lspci | egrep -i 'vga|3d|displejs'





Šajā gadījumā manā datorā ir instalēts NVIDIA GTX 1050 Ti GPU. Visticamāk, jūsu datorā būs instalēts cits NVIDIA GPU.


Tāpat pārliecinieties, vai jūsu Ubuntu 22.04 operētājsistēmā ir instalēti oficiālie NVIDIA draiveri ar šādu komandu:



$ lsmod | saķere nvidia

Ja jūsu Ubuntu 22.04 operētājsistēmā ir instalēti oficiālie NVIDIA draiveri, jūs redzēsit šādas izvades.


Tāpat pārbaudiet, vai oficiālie NVIDIA GPU draiveri darbojas ar šādu komandu:

$ nvidia-smi

Ja oficiālie NVIDIA GPU draiveri darbojas, jūs redzēsit šādas izejas. Kā redzat, manā Ubuntu 22.04 datorā ir instalēta oficiālā NVIDIA GPU draivera versija 525.78.01.


Ja jūsu Ubuntu 22.04 iekārtā nav instalēti oficiālie NVIDIA GPU draiveri un jums ir nepieciešama palīdzība, lūdzu, skatiet rakstu .

Docker CE atkarību instalēšana Ubuntu 22.04 LTS:

Lai varētu instalēt Docker CE Ubuntu 22.04, jums būs jāinstalē nepieciešamās Docker CE atkarības pakotnes Ubuntu 22.04.

Vispirms atjauniniet APT pakotnes repozitorija kešatmiņu ar šādu komandu:

$ sudo piemērots atjauninājums

APT pakotnes repozitorija kešatmiņa ir jāatjaunina.


Lai instalētu nepieciešamās Docker CE atkarības pakotnes, palaidiet šo komandu:

$ sudo apt-get install ca-certificates curl gnupg lsb-release

Lai apstiprinātu instalēšanu, nospiediet UN un pēc tam nospiediet .


Jāinstalē nepieciešamās Docker CE atkarības pakotnes.

Docker CE GPG atslēgu instalēšana Ubuntu 22.04 LTS:

Šajā sadaļā es jums parādīšu, kā instalēt Docker CE pakotņu krātuves GPG atslēgu Ubuntu 22.04.

Vispirms izveidojiet jaunu mapi /etc/apt/keyrings ar šādu komandu:

$ sudo mkdir -lpp / utt / apt / atslēgu riņķi

Lai instalētu Docker CE pakotņu repozitorija GPG atslēgu Ubuntu 22.04, palaidiet šo komandu:

$ čokurošanās -fsSL https: // download.docker.com / Linux / ubuntu / gpg | sudo gpg --dārgais -O / utt / apt / atslēgu riņķi / docker.gpg

Docker CE repozitorija instalēšana Ubuntu 22.04 LTS:

Lai instalētu Docker CE pakotņu repozitoriju Ubuntu 22.04, palaidiet šo komandu:

$ atbalss 'deb [arch= $(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stabils' | sudo tee / utt / apt / sources.list.d / docker.list > / izstrādātājs / null

Atjauniniet APT pakotnes repozitorija kešatmiņu, lai izmaiņas stātos spēkā.

$ sudo piemērots atjauninājums


Docker CE instalēšana Ubuntu 22.04 LTS:

Lai instalētu jaunāko Docker CE versiju Ubuntu 22.04, palaidiet šo komandu:

$ sudo apt-get install docker-ce docker-ce-cli containerd.io docker-compose-plugin

Lai apstiprinātu instalēšanu, nospiediet UN un pēc tam nospiediet .


Notiek Docker CE un vajadzīgās atkarības pakotņu lejupielāde. Tas prasīs kādu laiku, lai pabeigtu.


Tiek instalēts Docker CE un nepieciešamās atkarības pakotnes. Tas prasīs kādu laiku, lai pabeigtu.


Šajā brīdī ir jāinstalē Docker CE un nepieciešamās atkarības pakotnes.

Ubuntu 22.04 LTS pieteikšanās lietotāja pievienošana dokera grupai:

Lai izveidotu Docker konteinerus un pārvaldītu tos, neizmantojot sudo vai piesakoties kā root lietotājs, jums būs jāpievieno savs pieteikšanās lietotājs dokeris grupai.

Lai pievienotu Ubuntu 22.04 pieteikšanās lietotāju dokeris grupai, palaidiet šādu komandu:

$ sudo usermod -aG dokeris $ ( kas es esmu )

Lai izmaiņas stātos spēkā, restartējiet datoru ar šādu komandu:

$ sudo pārstartēt

Pārbaude, vai Docker CE ir pareizi instalēts Ubuntu 22.04 LTS:

Kad dators ir startēts, palaidiet šo komandu, lai pārbaudītu, vai varat piekļūt Docker bez superlietotāja privilēģijām.

$ docker versija

Ja viss darbojas, jums vajadzētu redzēt šādas izejas.

Kā redzat, es izmantoju Docker versiju 20.10.23 — jaunāko Docker CE versiju šī rakstīšanas laikā.

NVIDIA Container Toolkit GPG taustiņu instalēšana Ubuntu 22.04:

Šajā sadaļā es jums parādīšu, kā instalēt NVIDIA Container Toolkit pakotņu krātuves GPG atslēgu Ubuntu 22.04.

Lai instalētu NVIDIA Container Toolkit pakotņu krātuves GPG atslēgu Ubuntu 22.04, palaidiet šo komandu:

$ čokurošanās -fsSL https: // nvidia.github.io / libnvidia-konteiners / gpgkey | sudo gpg --dārgais -O / usr / dalīties / atslēgu riņķi / nvidia-container-toolkit-keyring.gpg

NVIDIA Container Toolkit repozitorija instalēšana Ubuntu 22.04 LTS:

Šajā sadaļā es jums parādīšu, kā instalēt NVIDIA Container Toolkit pakotnes repozitoriju Ubuntu 22.04.

Vispirms izveidojiet jaunu APT avota failu nvidia-container-toolkit.list iekš /etc/apt/sources.list.d/ direktoriju šādi:

$ sudo nano / utt / apt / sources.list.d / nvidia-container-toolkit.list

Iekš nvidia-container-toolkit.list failu, pievienojiet šādu rindiņu un nospiediet + X seko UN un lai saglabātu failu.

ka [ parakstījis= / usr / dalīties / atslēgu riņķi / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-konteiners / stabils / bezmaksas 22.04 / $ ( ARCH ) /



Ja izmantojat Ubuntu 20.04 LTS, viss, kas jums jādara, ir jāpievieno šāda rindiņa nvidia-container-toolkit.list faila vietā, un visam pārējam vajadzētu darboties bez izmaiņām.

ka [ parakstījis= / usr / dalīties / atslēgu riņķi / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-konteiners / stabils / bezmaksas 20.04 / $ ( ARCH ) /


Ja izmantojat Ubuntu 18.04 LTS, viss, kas jums jādara, ir jāpievieno šāda rindiņa nvidia-container-toolkit.list faila vietā, un visam pārējam vajadzētu darboties bez izmaiņām.

ka [ parakstījis= / usr / dalīties / atslēgu riņķi / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-konteiners / stabils / bezmaksas 18.04 / $ ( ARCH ) /


Lai izmaiņas stātos spēkā, atjauniniet APT pakotnes repozitorija kešatmiņu ar šādu komandu:

$ sudo piemērots atjauninājums


Nvidia-docker draiveru instalēšana Ubuntu 22.04 LTS:

Lai instalētu NVIDIA Docker draiverus Ubuntu 22.04, palaidiet šo komandu:

$ sudo apt uzstādīt nvidia-docker2

Lai apstiprinātu instalēšanu, nospiediet UN un pēc tam nospiediet .


Jāinstalē NVIDIA doka draiveri.


Lai izmaiņas stātos spēkā, restartējiet datoru ar šādu komandu:

$ sudo pārstartēt

Pārbaude, vai NVIDIA GPU ir pieejams no Docker konteineriem Ubuntu 22.04 LTS:

Šajā sadaļā es jums parādīšu, kā izveidot NVIDIA CUDA Docker konteiners un pārbaudiet, vai konteiners var piekļūt NVIDIA GPU no jūsu datora.

Lai izveidotu NVIDIA CUDA 12 Docker konteineru, pamatojoties uz Ubuntu 20.04 LTS, un palaist nvidia-smi Kad tā ir izveidota, lai pārbaudītu, vai tā var piekļūt NVIDIA GPU no jūsu datora, palaidiet šo komandu:

$ dokera skrējiens --rm --gpus visas nvidia / cuda:12.0.0-base-ubuntu20.04 nvidia-smi

Docker velk nvidia/cuda: 12.0.0-base-ubuntu20.04 attēls no Docker Hub. Tas prasīs kādu laiku, lai pabeigtu.


Kad NVIDIA CUDA doka attēls ir izvilkts un konteiners ir izveidots, nvidia-smi komanda darbosies tajā un izdrukās izvadi konsolē, kā redzat tālāk esošajā ekrānuzņēmumā.

Docker konteiners izmanto NVIDIA GPU draiveris 525.78.01 [1] un CUDA versija 12.0 [2] . Ja redzat līdzīgas izejas, Docker konteiners var piekļūt jūsu datora NVIDIA GPU.


Ja jums ir jāizmanto vecākas CUDA versijas, pārbaudiet .

Secinājums:

Šajā rakstā es jums parādīju, kā iestatīt Docker CE pakotnes repozitoriju Ubuntu 22.04. Esmu arī parādījis, kā instalēt jaunāko Docker CE versiju Ubuntu 22.04. Es jums parādīju, kā instalēt NVIDIA Container Toolkit pakotņu repozitoriju Ubuntu 22.04, kā arī kā instalēt NVIDIA Docker draiveri Ubuntu 22.04. Visbeidzot, es jums parādīju, kā piekļūt datora NVIDIA GPU no Docker konteinera.

Atsauces: