Kako koristiti NVIDIA GPU u Docker kontejnerima na Ubuntu 22.04 LTS

Kako Koristiti Nvidia Gpu U Docker Kontejnerima Na Ubuntu 22 04 Lts



Možete proći kroz svoj NVIDIA GPU u Docker spremnicima i pokrenuti CUDA programe na svom NVIDIA GPU-u iz ovih Docker spremnika. Ovo je vrlo korisna značajka za učenje AI (umjetne inteligencije). Mogućnost pokretanja AI kodova (tj. Tensorflow) na Docker spremnicima uštedjet će vam mnogo vremena. Također možete isprobati svoje AI kodove na različitim verzijama CUDA bez potrebe za promjenom verzije CUDA na vašem računalu.

Petljanje s NVIDIA drajverima i CUDA verzijama na vašem računalu može uzrokovati probleme poput NVIDIA drajvera koji ne rade ili vam ostavljaju crni/plavi ekran smrti. Dakle, uvijek je dobra ideja ostaviti svoj sustav onakvim kakav jest i napraviti promjene u izoliranom okruženju poput Docker spremnika. Ovo održava vaše glavno računalo čistim (od nepotrebnih razvojnih alata).







U ovom ću vam članku pokazati kako postaviti Docker CE i NVIDIA docker na Ubuntu 22.04 LTS tako da možete pristupiti NVIDIA GPU-u svog računala iz Docker spremnika i pokrenuti CUDA programe na svom NVIDIA GPU-u.



Sadržaj:

  1. Provjera jesu li službeni NVIDIA GPU upravljački programi instalirani na Ubuntu 22.04
  2. Instaliranje Docker CE ovisnosti na Ubuntu 22.04
  3. Instaliranje Docker CE GPG ključeva na Ubuntu 22.04
  4. Instaliranje Docker CE repozitorija na Ubuntu 22.04
  5. Instaliranje Docker CE na Ubuntu 22.04
  6. Dodavanje korisnika za prijavu na Ubuntu 22.04 LTS u docker grupu
  7. Provjera je li Docker CE ispravno instaliran na Ubuntu 22.04
  8. Instaliranje GPG ključeva NVIDIA Container Toolkit na Ubuntu 22.04
  9. Instaliranje repozitorija NVIDIA Container Toolkit na Ubuntu 22.04
  10. Instaliranje upravljačkih programa nvidia-docker na Ubuntu 22.04
  11. Provjera je li NVIDIA GPU dostupan iz Docker kontejnera u Ubuntu 22.04
  12. Zaključak
  13. Reference

Provjera jesu li službeni NVIDIA GPU upravljački programi instalirani na Ubuntu 22.04 LTS:

Prije nego što počnete, pokrenite sljedeću naredbu kako biste provjerili imate li NVIDIA GPU instaliran na vašem računalu.



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





U ovom slučaju, na računalu imam instaliran NVIDIA GTX 1050 Ti GPU. Najvjerojatnije ćete na svom računalu imati instaliran drugi NVIDIA GPU.


Također provjerite imate li službene NVIDIA upravljačke programe instalirane na vašem Ubuntu 22.04 operativnom sustavu pomoću sljedeće naredbe:



$ lsmod | zahvat nvidia

Ako su službeni NVIDIA upravljački programi instalirani na vašem Ubuntu 22.04 operativnom sustavu, vidjet ćete sljedeće rezultate.


Također provjerite rade li službeni NVIDIA GPU upravljački programi pomoću sljedeće naredbe:

$ nvidia-smi

Ako službeni NVIDIA GPU upravljački programi rade, vidjet ćete sljedeće rezultate. Kao što vidite, imam službeni NVIDIA GPU drajver verzije 525.78.01 instaliran na svom Ubuntu 22.04 računalu.


Ako nemate instalirane službene NVIDIA GPU upravljačke programe na vašem Ubuntu 22.04 računalu i potrebna vam je pomoć s tim, pogledajte članak .

Instaliranje Docker CE ovisnosti na Ubuntu 22.04 LTS:

Prije nego što možete instalirati Docker CE na Ubuntu 22.04, morat ćete instalirati potrebne pakete ovisnosti Docker CE na Ubuntu 22.04.

Najprije ažurirajte predmemoriju repozitorija APT paketa sljedećom naredbom:

$ sudo prikladno ažuriranje

Predmemoriju repozitorija APT paketa treba ažurirati.


Da biste instalirali potrebne pakete ovisnosti Docker CE, pokrenite sljedeću naredbu:

$ sudo apt-get instalacija ca-certifikati curl gnupg lsb-izdanje

Za potvrdu instalacije pritisnite I a zatim pritisnite .


Potrebni paketi ovisnosti Docker CE trebaju biti instalirani.

Instaliranje Docker CE GPG ključeva na Ubuntu 22.04 LTS:

U ovom odjeljku pokazat ću vam kako instalirati GPG ključ repozitorija paketa Docker CE na Ubuntu 22.04.

Najprije stvorite novu mapu /etc/apt/keyrings sa sljedećom naredbom:

$ sudo mkdir -str / itd / prikladan / Privjesci

Da biste instalirali GPG ključ repozitorija paketa Docker CE na Ubuntu 22.04, pokrenite sljedeću naredbu:

$ kovrča -fsSL https: // download.docker.com / linux / ubuntu / gpg | sudo gpg --drago -O / itd / prikladan / Privjesci / docker.gpg

Instaliranje Docker CE repozitorija na Ubuntu 22.04 LTS:

Da biste instalirali repozitorij paketa Docker CE na Ubuntu 22.04, pokrenite sljedeću naredbu:

$ jeka 'deb [arh= $(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stabilan' | sudo tee / itd / prikladan / izvori.list.d / docker.list > / dev / ništavan

Ažurirajte predmemoriju repozitorija APT paketa kako bi promjene stupile na snagu.

$ sudo prikladno ažuriranje


Instaliranje Docker CE na Ubuntu 22.04 LTS:

Da biste instalirali najnoviju verziju Docker CE na Ubuntu 22.04, pokrenite sljedeću naredbu:

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

Za potvrdu instalacije pritisnite I a zatim pritisnite .


Docker CE i potrebni paketi ovisnosti se preuzimaju. Trebat će neko vrijeme da se završi.


Docker CE i potrebni paketi ovisnosti se instaliraju. Trebat će neko vrijeme da se završi.


Docker CE i potrebni paketi ovisnosti trebaju biti instalirani u ovom trenutku.

Dodavanje korisnika za prijavu na Ubuntu 22.04 LTS u docker grupu:

Da biste izradili Docker spremnike i upravljali njima bez upotrebe sudo ili prijavljeni kao root korisnik, morat ćete dodati svog korisnika za prijavu u lučki radnik skupina.

Za dodavanje korisnika za prijavu vašeg Ubuntu 22.04 u lučki radnik grupe, pokrenite sljedeću naredbu:

$ sudo korisnički mod -aG doker $ ( tko sam ja )

Da bi promjene stupile na snagu, ponovno pokrenite računalo sljedećom naredbom:

$ sudo ponovno podizanje sustava

Provjera je li Docker CE ispravno instaliran na Ubuntu 22.04 LTS:

Nakon što se vaše računalo pokrene, pokrenite sljedeću naredbu da provjerite možete li pristupiti Dockeru bez privilegija superkorisnika.

$ docker verzija

Ako sve radi, trebali biste vidjeti sljedeće rezultate.

Kao što vidite, koristim Docker verziju 20.10.23 – najnoviju verziju Docker CE u vrijeme pisanja ovog teksta.

Instaliranje GPG ključeva NVIDIA Container Toolkit na Ubuntu 22.04:

U ovom odjeljku pokazat ću vam kako instalirati GPG ključ repozitorija paketa NVIDIA Container Toolkit na Ubuntu 22.04.

Da biste instalirali GPG ključ repozitorija paketa NVIDIA Container Toolkit na Ubuntu 22.04, pokrenite sljedeću naredbu:

$ kovrča -fsSL https: // nvidia.github.io / libnvidia-spremnik / gpgkey | sudo gpg --drago -O / usr / udio / Privjesci / nvidia-container-toolkit-keyring.gpg

Instaliranje repozitorija NVIDIA Container Toolkit na Ubuntu 22.04 LTS:

U ovom odjeljku, pokazat ću vam kako instalirati repozitorij paketa NVIDIA Container Toolkit na Ubuntu 22.04.

Prvo stvorite novu APT izvornu datoteku nvidia-container-toolkit.list u /etc/apt/sources.list.d/ imenik kako slijedi:

$ sudo nano / itd / prikladan / izvori.list.d / nvidia-container-toolkit.list

u nvidia-container-toolkit.list datoteku, dodajte sljedeći redak i pritisnite + x slijedi I i za spremanje datoteke.

da [ potpisao= / usr / udio / Privjesci / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-spremnik / stabilan / besplatno22.04 / $ ( ARCH ) /



Ako koristite Ubuntu 20.04 LTS, sve što trebate učiniti je dodati sljedeći redak u nvidia-container-toolkit.list datoteku i sve ostalo bi trebalo raditi nepromijenjeno.

da [ potpisao= / usr / udio / Privjesci / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-spremnik / stabilan / besplatno20.04 / $ ( ARCH ) /


Ako koristite Ubuntu 18.04 LTS, sve što trebate učiniti je dodati sljedeći redak u nvidia-container-toolkit.list datoteku i sve ostalo bi trebalo raditi nepromijenjeno.

da [ potpisao= / usr / udio / Privjesci / nvidia-container-toolkit-keyring.gpg ] https: // nvidia.github.io / libnvidia-spremnik / stabilan / besplatno18.04 / $ ( ARCH ) /


Da bi promjene stupile na snagu, ažurirajte predmemoriju repozitorija APT paketa sljedećom naredbom:

$ sudo prikladno ažuriranje


Instaliranje upravljačkih programa nvidia-docker na Ubuntu 22.04 LTS:

Da biste instalirali upravljačke programe NVIDIA Docker na Ubuntu 22.04, pokrenite sljedeću naredbu:

$ sudo prikladan instalirati nvidia-docker2

Za potvrdu instalacije pritisnite I a zatim pritisnite .


NVIDIA docker upravljački programi trebaju biti instalirani.


Da bi promjene stupile na snagu, ponovno pokrenite računalo sljedećom naredbom:

$ sudo ponovno podizanje sustava

Provjera je li NVIDIA GPU dostupan iz Docker kontejnera u Ubuntu 22.04 LTS:

U ovom odjeljku pokazat ću vam kako izraditi NVIDIA CUDA Docker spremnik i provjerite može li spremnik pristupiti NVIDIA GPU-u s vašeg računala.

Za izradu NVIDIA CUDA 12 Docker spremnika temeljenog na Ubuntu 20.04 LTS i pokretanje nvidia-smi naredbu u njoj nakon što je stvorena kako biste provjerili može li pristupiti NVIDIA GPU-u s vašeg računala, pokrenite sljedeću naredbu:

$ docker run --rm --gpus sve nvidia / cuda:12.0.0-base-ubuntu20.04 nvidia-smi

Docker povlači nvidia/cuda:12.0.0-base-ubuntu20.04 slika iz Docker Huba. Trebat će neko vrijeme da se završi.


Nakon što se povuče NVIDIA CUDA docker slika i stvori spremnik, nvidia-smi naredba će se pokrenuti na njemu i ispisati izlaz na konzoli kao što možete vidjeti na slici ispod.

Docker spremnik koristi NVIDIA GPU drajver 525.78.01 [1] i CUDA verzija 12.0 [2] . Ako vidite slične rezultate, Docker spremnik može pristupiti NVIDIA GPU-u vašeg računala.


Ako trebate koristiti starije verzije CUDA-e, provjerite .

Zaključak:

U ovom članku sam vam pokazao kako postaviti repozitorij paketa Docker CE na Ubuntu 22.04. Također sam vam pokazao kako instalirati najnoviju verziju Docker CE na Ubuntu 22.04. Pokazao sam vam kako instalirati repozitorij paketa NVIDIA Container Toolkit na Ubuntu 22.04 kao i kako instalirati NVIDIA Docker upravljački program na Ubuntu 22.04. Konačno, pokazao sam vam kako pristupiti NVIDIA GPU-u vašeg računala iz Docker spremnika.

Reference: