NVIDIA GPU:n käyttäminen Docker Containersissa Ubuntu 22.04 LTS:ssä

Nvidia Gpu N Kayttaminen Docker Containersissa Ubuntu 22 04 Lts Ssa



Voit siirtää NVIDIA-grafiikkasuorittimesi Docker-säilöissä ja suorittaa CUDA-ohjelmia NVIDIA-grafiikkasuorittimessasi näistä Docker-säiliöistä. Tämä on erittäin hyödyllinen ominaisuus tekoälyn (AI) oppimiseen. AI-koodien (eli Tensorflow) suorittaminen Docker-säiliöissä säästää paljon aikaa. Voit myös kokeilla tekoälykoodejasi eri CUDA-versioissa ilman, että sinun tarvitsee vaihtaa CUDA-versiota tietokoneellasi.

Tietokoneen NVIDIA-ajureiden ja CUDA-versioiden näpertely voi aiheuttaa ongelmia, kuten NVIDIA-ajureiden toimimattomuuden tai mustan/sinisen ruudun. Joten on aina hyvä idea jättää järjestelmä ennalleen ja tehdä muutoksia eristettyyn ympäristöön, kuten Docker-säiliöihin. Tämä pitää päätietokoneesi puhtaana (tarpeettomista kehitystyökaluista).







Tässä artikkelissa aion näyttää sinulle, kuinka Docker CE ja NVIDIA Docker asennetaan Ubuntu 22.04 LTS:ään, jotta voit käyttää tietokoneesi NVIDIA GPU:ta Docker-säiliöistä ja suorittaa CUDA-ohjelmia NVIDIA GPU:ssa.



Sisällysluettelo:

  1. Tarkistamme, onko viralliset NVIDIA GPU -ohjaimet asennettu Ubuntu 22.04:ään
  2. Docker CE -riippuvuuksien asentaminen Ubuntuun 22.04
  3. Docker CE GPG Keys -avaimien asentaminen Ubuntuun 22.04
  4. Docker CE Repositoryn asentaminen Ubuntuun 22.04
  5. Docker CE:n asentaminen Ubuntuun 22.04
  6. Ubuntu 22.04 LTS -kirjautumiskäyttäjän lisääminen telakointiryhmään
  7. Tarkistetaan, onko Docker CE asennettu oikein Ubuntuun 22.04
  8. NVIDIA Container Toolkit GPG Keys -avaimien asentaminen Ubuntu 22.04:ään
  9. NVIDIA Container Toolkit Repositoryn asentaminen Ubuntuun 22.04
  10. Nvidia-docker-ajurien asentaminen Ubuntuun 22.04
  11. Tarkistamme, onko NVIDIA GPU käytettävissä Docker Containersista Ubuntu 22.04:ssä
  12. Johtopäätös
  13. Viitteet

Tarkistaminen, onko viralliset NVIDIA GPU -ohjaimet asennettu Ubuntu 22.04 LTS:ään:

Ennen kuin aloitat, suorita seuraava komento varmistaaksesi, että tietokoneellesi on asennettu NVIDIA GPU.



$ lspci | egrep -i 'vga|3d|näyttö'





Tässä tapauksessa tietokoneelleni on asennettu NVIDIA GTX 1050 Ti GPU. Tietokoneellesi on todennäköisesti asennettu toinen NVIDIA GPU.


Varmista myös, että sinulla on viralliset NVIDIA-ohjaimet asennettuna Ubuntu 22.04 -käyttöjärjestelmääsi seuraavalla komennolla:



$ lsmod | ote nvidia

Jos viralliset NVIDIA-ohjaimet on asennettu Ubuntu 22.04 -käyttöjärjestelmääsi, näet seuraavat lähdöt.


Tarkista myös, toimivatko viralliset NVIDIA GPU -ohjaimet seuraavan komennon kanssa:

$ nvidia-smi

Jos viralliset NVIDIA GPU -ohjaimet toimivat, näet seuraavat lähdöt. Kuten näet, minulla on virallinen NVIDIA GPU Driver -versio 525.78.01 asennettuna Ubuntu 22.04 -koneeseeni.


Jos sinulla ei ole virallisia NVIDIA GPU -ohjaimia asennettuna Ubuntu 22.04 -koneellesi ja tarvitset apua sen kanssa, katso artikkeli .

Docker CE -riippuvuuksien asentaminen Ubuntu 22.04 LTS:ään:

Ennen kuin voit asentaa Docker CE:n Ubuntu 22.04:ään, sinun on asennettava tarvittavat Docker CE -riippuvuuspaketit Ubuntu 22.04:ään.

Päivitä ensin APT-pakettivaraston välimuisti seuraavalla komennolla:

$ sudo osuva päivitys

APT-pakettivaraston välimuisti tulee päivittää.


Asenna Docker CE:n tarvittavat riippuvuuspaketit suorittamalla seuraava komento:

$ sudo apt-get install ca-sertifikaatit curl gnupg lsb-julkaisu

Vahvista asennus painamalla JA ja paina sitten .


Tarvittavat Docker CE -riippuvuuspaketit tulee asentaa.

Docker CE GPG -avaimien asentaminen Ubuntu 22.04 LTS:ään:

Tässä osiossa aion näyttää sinulle, kuinka Docker CE -pakettivaraston GPG-avain asennetaan Ubuntu 22.04:ään.

Luo ensin uusi kansio /etc/apt/keyrings seuraavalla komennolla:

$ sudo mkdir -s / jne / apt / avaimenperät

Asenna Docker CE -pakettivaraston GPG-avain Ubuntu 22.04:ään suorittamalla seuraava komento:

$ kiemura -fsSL https: // download.docker.com / Linux / ubuntu / gpg | sudo gpg --rakas -O / jne / apt / avaimenperät / docker.gpg

Docker CE -varaston asentaminen Ubuntu 22.04 LTS:ään:

Asenna Docker CE -pakettivarasto Ubuntu 22.04:ään suorittamalla seuraava komento:

$ kaiku 'deb [arch= $(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) vakaa' | sudo tee / jne / apt / sources.list.d / docker.list > / kehittäjä / tyhjä

Päivitä APT-pakettivaraston välimuisti, jotta muutokset tulevat voimaan.

$ sudo osuva päivitys


Docker CE:n asentaminen Ubuntu 22.04 LTS:ään:

Asenna Docker CE:n uusin versio Ubuntu 22.04:ään suorittamalla seuraava komento:

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

Vahvista asennus painamalla JA ja paina sitten .


Docker CE:tä ja tarvittavia riippuvuuspaketteja ladataan. Sen valmistuminen kestää jonkin aikaa.


Docker CE:tä ja tarvittavia riippuvuuspaketteja asennetaan. Sen valmistuminen kestää jonkin aikaa.


Docker CE ja tarvittavat riippuvuuspaketit tulee asentaa tässä vaiheessa.

Ubuntu 22.04 LTS -kirjautumiskäyttäjän lisääminen telakointiryhmään:

Jos haluat luoda Docker-säilöjä ja hallita niitä ilman sudoa tai kirjautumatta sisään pääkäyttäjänä, sinun on lisättävä kirjautumiskäyttäjäsi satamatyöläinen ryhmä.

Voit lisätä Ubuntu 22.04:n kirjautumiskäyttäjän satamatyöläinen ryhmä, suorita seuraava komento:

$ sudo usermod -aG telakka $ ( kuka olen )

Jotta muutokset tulevat voimaan, käynnistä tietokoneesi uudelleen seuraavalla komennolla:

$ sudo käynnistä uudelleen

Tarkistaminen, onko Docker CE asennettu oikein Ubuntu 22.04 LTS:ään:

Kun tietokone käynnistyy, suorita seuraava komento varmistaaksesi, että voit käyttää Dockeria ilman pääkäyttäjän oikeuksia.

$ Docker-versio

Jos kaikki toimii, sinun pitäisi nähdä seuraavat lähdöt.

Kuten näet, käytän Dockerin versiota 20.10.23 – viimeisintä Docker CE:n versiota tätä kirjoitettaessa.

NVIDIA Container Toolkit GPG Keys -avaimien asentaminen Ubuntu 22.04:ään:

Tässä osiossa aion näyttää sinulle, kuinka NVIDIA Container Toolkit -pakettivaraston GPG-avain asennetaan Ubuntu 22.04:ään.

Asenna NVIDIA Container Toolkit -pakettivaraston GPG-avain Ubuntu 22.04:ään suorittamalla seuraava komento:

$ kiemura -fsSL https: // nvidia.github.io / libnvidia-säilö / gpgkey | sudo gpg --rakas -O / usr / Jaa / avaimenperät / nvidia-container-toolkit-keyring.gpg

NVIDIA Container Toolkit -varaston asentaminen Ubuntu 22.04 LTS:ään:

Tässä osiossa aion näyttää sinulle, kuinka NVIDIA Container Toolkit -pakettivarasto asennetaan Ubuntu 22.04:ään.

Luo ensin uusi APT-lähdetiedosto nvidia-container-toolkit.list in /etc/apt/sources.list.d/ hakemisto seuraavasti:

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

Vuonna nvidia-container-toolkit.list tiedosto, lisää seuraava rivi ja paina + X jonka jälkeen JA ja tallentaaksesi tiedoston.

että [ allekirjoittanut = / usr / Jaa / avaimenperät / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-säilö / vakaa / ilmainen 22.04 / $ ( KAARI ) /



Jos käytät Ubuntu 20.04 LTS:ää, sinun tarvitsee vain lisätä seuraava rivi nvidia-container-toolkit.list tiedosto ja kaiken muun pitäisi toimia ennallaan.

että [ allekirjoittanut = / usr / Jaa / avaimenperät / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-säilö / vakaa / ilmainen 20.04 / $ ( KAARI ) /


Jos käytät Ubuntu 18.04 LTS:ää, sinun tarvitsee vain lisätä seuraava rivi nvidia-container-toolkit.list tiedosto ja kaiken muun pitäisi toimia ennallaan.

että [ allekirjoittanut = / usr / Jaa / avaimenperät / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-säilö / vakaa / ilmainen 18.04 / $ ( KAARI ) /


Jotta muutokset tulevat voimaan, päivitä APT-pakettivaraston välimuisti seuraavalla komennolla:

$ sudo osuva päivitys


Nvidia-docker-ajurien asentaminen Ubuntu 22.04 LTS:ään:

Asenna NVIDIA Docker -ohjaimet Ubuntu 22.04:ään suorittamalla seuraava komento:

$ sudo apt Asentaa nvidia-docker2

Vahvista asennus painamalla JA ja paina sitten .


NVIDIA-telakointiaseman ohjaimet tulee asentaa.


Jotta muutokset tulevat voimaan, käynnistä tietokoneesi uudelleen seuraavalla komennolla:

$ sudo käynnistä uudelleen

Tarkistaminen, onko NVIDIA GPU käytettävissä Docker Containersista Ubuntu 22.04 LTS:ssä:

Tässä osiossa aion näyttää sinulle, kuinka luodaan NVIDIA CUDA Docker -kontti ja varmista, että säilö voi käyttää NVIDIA GPU:ta tietokoneeltasi.

Luodaksesi Ubuntu 20.04 LTS:ään perustuvan NVIDIA CUDA 12 Docker -säiliön ja suorittamalla nvidia-smi komento siinä, kun se on luotu varmistaaksesi, voiko se käyttää NVIDIA GPU:ta tietokoneeltasi, suorita seuraava komento:

$ telakkajuoksu --rm --gpus kaikki nvidia / cuda:12.0.0-base-ubuntu20.04 nvidia-smi

Docker vetää nvidia/cuda: 12.0.0-base-ubuntu20.04 kuva Docker Hubista. Sen valmistuminen kestää jonkin aikaa.


Kun NVIDIA CUDA Docker -kuva on vedetty ja säilö on luotu, nvidia-smi komento suoritetaan siinä ja tulostaa tulosteen konsoliin, kuten alla olevasta kuvakaappauksesta näet.

Docker-säiliö käyttää NVIDIA GPU-ohjain 525.78.01 [1] ja CUDA versio 12.0 [2] . Jos näet samanlaisia ​​lähtöjä, Docker-säilö voi käyttää tietokoneesi NVIDIA GPU:ta.


Jos sinun on käytettävä CUDA:n vanhempia versioita, tarkista .

Johtopäätös:

Tässä artikkelissa olen näyttänyt sinulle, kuinka Docker CE -pakettivarasto määritetään Ubuntu 22.04:ssä. Olen myös näyttänyt sinulle, kuinka Docker CE:n uusin versio asennetaan Ubuntu 22.04:ään. Olen näyttänyt sinulle, kuinka NVIDIA Container Toolkit -pakettivarasto asennetaan Ubuntu 22.04:ään sekä NVIDIA Docker -ohjain Ubuntu 22.04:ään. Lopuksi olen näyttänyt sinulle, kuinka voit käyttää tietokoneesi NVIDIA GPU:ta Docker-säiliöstä.

Viitteet: