Hvordan bruke NVIDIA GPU i Docker -containere på Ubuntu Top 10.Topp 10 LTS

Du kan gjennomgå NVIDIA GPU i Docker -containere og kjøre CUDA -programmer på NVIDIA GPU fra disse Docker -containerne. Dette er en veldig nyttig funksjon for å lære AI (kunstig intelligens). Å kunne kjøre AI -koder (i.e. Tensorflow) på Docker -containere vil spare deg for mye tid. Du kan også prøve ut AI -kodene dine på forskjellige CUDA -versjoner uten å måtte endre CUDA -versjonen på datamaskinen din.

Fikling med NVIDIA -drivere og CUDA -versjoner på datamaskinen din kan forårsake problemer som NVIDIA -sjåførene som ikke fungerer eller etterlater deg en svart/blå dødskjerm. Så det er alltid en god idé å forlate systemet ditt som det er og gjøre endringer i et isolert miljø som Docker -containere. Dette holder din viktigste datamaskin ren (av unødvendige utviklingsverktøy).

I denne artikkelen skal jeg vise deg hvordan du konfigurerer Docker CE og Nvidia Docker på Ubuntu 22.04 LTS slik at du får tilgang til NVIDIA GPU på datamaskinen din fra Docker Containers og kjører CUDA -programmer på NVIDIA GPU.

Innholdsfortegnelse:

1. Kontroller om de offisielle NVIDIA GPU -driverne er installert på Ubuntu 22.04 LTS
2. Installere Docker CE -avhengigheter av Ubuntu 22.04 LTS
3. Installere Docker CE GPG -nøkler på Ubuntu 22.04 LTS
4. Installere Docker CE -depotet på Ubuntu 22.04 LTS
5. Installere Docker CE på Ubuntu 22.04 LTS
6. Legge til Ubuntu 22.04 LTS -påloggingsbruker til Docker Group
7. Kontroller om Docker CE ble installert riktig på Ubuntu 22.04 LTS
8. Installere NVIDIA Container Toolkit GPG -nøkler på Ubuntu 22.04 LTS
9. Installere NVIDIA Container Toolkit -depotet på Ubuntu 22.04 LTS
10. Installere Nvidia-Docker-drivere på Ubuntu 22.04 LTS
11. Kontroller om NVIDIA GPU er tilgjengelig fra Docker -containere i Ubuntu 22.04 LTS
12. Konklusjon
13. Referanser

Kontroller om de offisielle NVIDIA GPU -driverne er installert på Ubuntu 22.04 LTS:

Før du kommer i gang, kjør følgende kommando for å bekrefte at du har en NVIDIA GPU installert på datamaskinen din.

$ LSPCI | Egrep -i 'VGA | 3D | Display'

I dette tilfellet har jeg NVIDIA GTX 1050 TI GPU installert på datamaskinen min. Du vil mest sannsynlig ha en annen NVIDIA GPU installert på datamaskinen din.

Forsikre deg også om at du har de offisielle NVIDIA -driverne installert på Ubuntu 22.04 Operativsystem med følgende kommando:

$ lsmod | grep nvidia

Hvis de offisielle NVIDIA -driverne er installert på Ubuntu 22.04 Operativsystem, vil du se følgende utganger.

Kontroller også om de offisielle NVIDIA GPU -driverne jobber med følgende kommando:

$ nvidia-Smi

Hvis de offisielle NVIDIA GPU -sjåførene fungerer, vil du se følgende utganger. Som du ser, har jeg den offisielle NVIDIA GPU -driveren versjon 525.78.01 installert på Ubuntu 22.04 Maskin.

Hvis du ikke har de offisielle NVIDIA GPU -driverne installert på Ubuntu 22.04 Maskin og du trenger hjelp med det, vennligst sjekk artikkelen hvordan du installerer Nvidia -drivere på Ubuntu 22.04 LTS.

Installere Docker CE -avhengigheter av Ubuntu 22.04 LTS:

Før du kan installere Docker CE på Ubuntu 22.04, må du installere de nødvendige Docker CE -avhengighetspakker på Ubuntu 22.04.

Oppdater først Apt Package Repository Cache med følgende kommando:

$ sudo apt oppdatering

Apt Package Repository Cache skal oppdateres.

For å installere de nødvendige avhengighetspakkene med Docker CE, kjør følgende kommando:

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

For å bekrefte installasjonen, trykk Y Og trykk deretter .

De nødvendige Docker CE -avhengighetspakker skal installeres.

Installere Docker CE GPG -nøkler på Ubuntu 22.04 LTS:

I dette avsnittet skal jeg vise deg hvordan du installerer GPG -tasten til Docker CE -pakningsoppføret på Ubuntu 22.04.

Først må du opprette en ny mappe /etc/apt/nøkkelringer med følgende kommando:

$ sudo mkdir -p/etc/apt/nøkkelringer

For å installere GPG -tasten til Docker CE -pakningsoppføringen på Ubuntu 22.04, kjør følgende kommando:

$ curl -fssl https: // nedlasting.Docker.com/linux/ubuntu/gpg | sudo gpg - -dearmor -o/etc/apt/nøkkelringer/docker.GPG

Installere Docker CE -depotet på Ubuntu 22.04 LTS:

For å installere Docker CE -pakkelageret på Ubuntu 22.04, kjør følgende kommando:

$ ECHO "Deb [Arch = $ (DPKG-PRINTR-ARCHITECTUR) Signert-By =/etc/Apt/KeyRings/Docker.gpg] https: // nedlasting.Docker.com/linux/ubuntu $ (lsb_release -cs) stabil "| sudo tee/etc/apt/kilder.liste.D/Docker.Liste> /dev /null

Oppdater Apt Package Repository Cache for endringene som skal tre i kraft.

$ sudo apt oppdatering

Installere Docker CE på Ubuntu 22.04 LTS:

For å installere den siste versjonen av Docker CE på Ubuntu 22.04, kjør følgende kommando:

$ sudo apt-get install docker-ce docker-ce-cli containerd.IO Docker-Compose-plugin

For å bekrefte installasjonen, trykk Y Og trykk deretter .

Docker CE og de nødvendige avhengighetspakkene lastes ned. Det vil ta en stund å fullføre.

Docker CE og de nødvendige avhengighetspakker blir installert. Det vil ta en stund å fullføre.

Docker CE og de nødvendige avhengighetspakkene skal installeres på dette tidspunktet.

Legge til Ubuntu 22.04 LTS LOGIN BRUKER TIL DOCKER GROUP:

For å lage Docker -containere og administrere dem uten å bruke sudo eller logget inn som rotbruker, må du legge til påloggingsbrukeren din til Docker gruppe.

For å legge til påloggingsbrukeren til Ubuntu 22.04 til Docker gruppe, kjør følgende kommando:

$ sudo usermod -ag docker $ (whoami)

For at endringene skal tre i kraft, start datamaskinen på nytt med følgende kommando:

$ sudo omstart

Kontroller om Docker CE ble installert riktig på Ubuntu 22.04 LTS:

Når datamaskinen starter, kjør følgende kommando for å bekrefte at du kan få tilgang til Docker uten superbrukerrettigheter.

$ Docker -versjon

Hvis alt fungerer, bør du se følgende utganger.

Som du ser, kjører jeg Docker versjon 20.10.23 - Den siste versjonen av Docker CE på tidspunktet for dette forfatterskapet.

Installere NVIDIA Container Toolkit GPG -nøkler på Ubuntu 22.04:

I denne delen skal jeg vise deg hvordan du installerer GPG -tasten til NVIDIA Container Toolkit Package Repository på Ubuntu 22.04.

For å installere GPG -tasten til NVIDIA Container Toolkit Package Repository på Ubuntu 22.04, kjør følgende kommando:

$ curl -fssl https: // nvidia.github.io/libnvidia-container/gpgkey | sudo gpg--dearmor -o/usr/share/keyrings/nvidia-container-toolkit-keyring.GPG

Installere NVIDIA Container Toolkit -depotet på Ubuntu 22.04 LTS:

I denne delen skal jeg vise deg hvordan du installerer Nvidia Container Toolkit Package Repository på Ubuntu 22.04.

Opprett først en ny passende kildefil Nvidia-Container-Toolkit.liste i /etc/apt/kilder.liste.d/ Katalog som følger:

$ sudo nano/etc/apt/kilder.liste.D/NVIDIA-CONTAINER-TOUKKIT.liste

I Nvidia-Container-Toolkit.liste fil, legg til følgende linje og trykk + X etterfulgt av Y og For å lagre filen.

Deb [signert-by =/usr/share/nøkkelring/nvidia-container-toolkit-keyring.gpg] https: // nvidia.github.io/libnvidia-container/stabil/ubuntu22.04 /$ (Arch) /

MERK: Hvis du bruker Ubuntu 20.04 LTS, så alt du trenger å gjøre er å legge til følgende linje i Nvidia-Container-Toolkit.liste fil i stedet og alt annet skal fungere uendret.

Deb [signert-by =/usr/share/nøkkelring/nvidia-container-toolkit-keyring.gpg] https: // nvidia.github.IO/libnvidia-container/stabil/ubuntu20.04 /$ (Arch) /

MERK: Hvis du bruker Ubuntu 18.04 LTS, så alt du trenger å gjøre er å legge til følgende linje i Nvidia-Container-Toolkit.liste fil i stedet og alt annet skal fungere uendret.

Deb [signert-by =/usr/share/nøkkelring/nvidia-container-toolkit-keyring.gpg] https: // nvidia.github.io/libnvidia-container/stabil/ubuntu18.04 /$ (Arch) /

For at endringene skal tre i kraft, oppdater Apt Package Repository Cache med følgende kommando:

$ sudo apt oppdatering

Installere Nvidia-Docker-drivere på Ubuntu 22.04 LTS:

For å installere Nvidia Docker -driverne på Ubuntu 22.04, kjør følgende kommando:

$ sudo apt install nvidia-docker2

For å bekrefte installasjonen, trykk Y Og trykk deretter .

Nvidia Docker -driverne skal installeres.

For at endringene skal tre i kraft, start datamaskinen på nytt med følgende kommando:

$ sudo omstart

Kontroller om NVIDIA GPU er tilgjengelig fra Docker -containere i Ubuntu 22.04 LTS:

I denne delen skal jeg vise deg hvordan du oppretter en Nvidia Cuda Docker -beholder og bekreft at beholderen kan få tilgang til NVIDIA GPU fra datamaskinen din.

For å lage en Nvidia Cuda 12 Docker -beholder basert på Ubuntu 20.04 LTS og kjører Nvidia-Smi Kommando i den når den er opprettet for å bekrefte om den kan få tilgang til NVIDIA GPU fra datamaskinen din, kjør følgende kommando:

$ Docker Run - -RM -GPUS All Nvidia/CUDA: 12.0.0-Base-Ubuntu20.04 NVIDIA-SMI

Docker trekker NVIDIA/CUDA: 12.0.0-Base-Ubuntu20.04 Bilde fra Docker Hub. Det vil ta en stund å fullføre.

Når Nvidia Cuda Docker -bildet er trukket og en beholder er opprettet, er det Nvidia-Smi Kommandoen kjøres på den og skriver ut utgangen på konsollen som du kan se på skjermdumpen nedenfor.

Docker -beholderen bruker NVIDIA GPU Driver 525.78.01^[1] og CUDA versjon 12.0^[2]. Hvis du ser lignende utganger, kan Docker -beholderen få tilgang til NVIDIA GPU på datamaskinen.

Hvis du trenger å bruke eldre versjoner av CUDA, sjekk NVIDIA/CUDA -bildet på Docker Hub.

Konklusjon:

I denne artikkelen har jeg vist deg hvordan du konfigurerer Docker CE -pakkelageret på Ubuntu 22.04. Jeg har også vist deg hvordan du installerer den nyeste versjonen av Docker CE på Ubuntu 22.04. Jeg har vist deg hvordan du installerer Nvidia Container Toolkit Package Repository på Ubuntu 22.04 samt hvordan du installerer Nvidia Docker -driveren på Ubuntu 22.04. Til slutt har jeg vist deg hvordan du får tilgang til NVIDIA GPU på datamaskinen din fra en Docker -beholder.

Referanser:

1. https: // dokumenter.nvidia.com/datasenter/sky-native/container-toolkit/install-guide.html
2. https: // hub.Docker.com/r/nvidia/cuda/tags