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 Containers. Dette holder din viktigste datamaskin ren (av unødvendige utviklingsverktøy).

I denne artikkelen skal jeg vise deg hvordan du installerer Docker CE og Nvidia Docker på Linux Mint 21, slik at du kan få tilgang til NVIDIA GPU på datamaskinen din fra Docker Containers og kjøre CUDA -programmer på NVIDIA GPU.

Innholdsfortegnelse:

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

Kontroller om de offisielle NVIDIA GPU -driverne er installert på Linux Mint 21

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

$ LSPCI | grep vga

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å Linux Mint 21 -operativsystemet med følgende kommando:

$ lsmod | grep nvidia

Hvis de offisielle NVIDIA -driverne er installert på Linux Mint 21 -operativsystemet, 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 vist har jeg den offisielle NVIDIA GPU -driveren versjon 525.78.01 installert på min Linux Mint 21 -maskin.

Hvis du ikke har de offisielle NVIDIA GPU -driverne installert på Linux Mint 21 -maskinen din, og du trenger hjelp med det, kan du sjekke artikkelen hvordan du installerer NVIDIA -drivere på Linux Mint 21.

Installere Docker CE -avhengigheter av Linux Mint 21

Før du kan installere Docker CE på Linux Mint 21, må du installere de nødvendige Docker CE -avhengighetspakker på Linux Mint 21.

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å Linux Mint 21

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

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 -pakkelageret på Linux Mint 21, 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 -depot på Linux Mint 21

For å installere Docker CE -pakningsreposten på Linux Mint 21, kjør følgende kommando:

$ ECHO "Deb [Arch = $ (DPKG-PRINTR-ARCHITECTUR) Signert-By =/etc/Apt/KeyRings/Docker.gpg] https: // nedlasting.Docker.com/linux/ubuntu jammy 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å Linux Mint 21

For å installere den nyeste versjonen av Docker CE på Linux Mint 21, 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 Linux Mint 21 Login -brukeren til Docker -gruppen

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 Linux Mint 21 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å Linux Mint 21

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 vist 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å Linux Mint 21

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

For å installere GPG -tasten til NVIDIA Container Toolkit Package Repository på Linux Mint 21, 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å Linux Mint 21

I denne delen skal jeg vise deg hvordan du installerer Nvidia Container Toolkit Package Repository på Linux Mint 21.

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) /

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

$ sudo apt oppdatering

Installere Nvidia-Docker-drivere på Linux Mint 21

For å installere Nvidia Docker -driverne på Linux Mint 21, 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 Linux Mint 21

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 det. Når den er opprettet for å bekrefte om den har tilgang til NVIDIA GPU fra datamaskinen din, kjører du 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, er det opprettet en beholder, Nvidia-Smi Kommandoen kjøres på den og skriver ut utgangen på konsollen som du kan se på skjermdumpen nedenfor.

Som vist bruker Docker -beholderen 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

Jeg har vist deg hvordan du installerer Docker CE -pakkelageret på Linux Mint 21, hvordan du installerer den nyeste versjonen av Docker CE på Linux Mint 21, hvordan installerer NVIDIA Container Toolkit Package Repository på Linux Mint 21, samt hvordan du installerer Nvidia Docker -sjåføren på Linux Mint 21. 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