Puoi passthrough la tua GPU Nvidia in contenitori Docker ed eseguire programmi CUDA sulla tua GPU NVIDIA da questi contenitori Docker. Questa è una caratteristica molto utile per l'apprendimento dell'IA (intelligenza artificiale). Essere in grado di eseguire codici AI (i.e. Tensorflow) su contenitori Docker ti farà risparmiare molto tempo. Puoi anche provare i tuoi codici AI su diverse versioni CUDA senza dover cambiare la versione CUDA sul tuo computer.

Presero con driver Nvidia e versioni CUDA sul tuo computer può causare problemi come i driver Nvidia che non funzionano o lasciarti con uno schermo di morte nero/blu. Quindi, è sempre una buona idea lasciare il tuo sistema così com'è e apportare modifiche in un ambiente isolato come i contenitori Docker. Questo mantiene il tuo computer principale pulito (di strumenti di sviluppo inutili).

In questo articolo, ti mostrerò come installare Docker CE e Nvidia Docker su Linux Mint 21 in modo da poter accedere alla GPU NVIDIA del tuo computer dai contenitori Docker ed eseguire programmi CUDA sulla tua GPU NVIDIA.

Sommario:

1. Verifica se i driver GPU NVIDIA ufficiali sono installati su Linux Mint 21
2. Installazione delle dipendenze Docker CE da Linux Mint 21
3. Installazione dei tasti GPG Docker CE su Linux Mint 21
4. Installazione del repository Docker CE su Linux Mint 21
5. Installazione di Docker CE su Linux Mint 21
6. Aggiunta dell'utente di accesso Linux Mint 21 al gruppo Docker
7. Verifica se Docker CE è stato installato correttamente su Linux Mint 21
8. Installazione dei tasti GPG per toolkit contenitore NVIDIA su Linux Mint 21
9. Installazione del repository di toolkit del contenitore Nvidia su Linux Mint 21
10. Installazione dei driver Nvidia-Docker su Linux Mint 21
11. Verifica se la GPU Nvidia è accessibile dai contenitori Docker in Linux Mint 21
12. Conclusione
13. Riferimenti

Verifica se i driver GPU NVIDIA ufficiali sono installati su Linux Mint 21

Prima di iniziare, eseguire il seguente comando per verificare di avere una GPU NVIDIA installata sul tuo computer:

$ lspci | grep vga

In questo caso, ho la GPU Nvidia GTX 1050 TI installata sul mio computer. Molto probabilmente avrai una GPU Nvidia diversa installata sul tuo computer.

Inoltre, assicurati di avere i driver Nvidia ufficiali installati sul tuo sistema operativo Linux Mint 21 con il seguente comando:

$ lsmod | grep nvidia

Se i driver Nvidia ufficiali sono installati sul sistema operativo Linux Mint 21, vedrai le seguenti uscite:

Inoltre, controlla se i driver GPU NVIDIA ufficiali stanno lavorando con il seguente comando:

$ nvidia-smi

Se i driver GPU Nvidia ufficiali funzionano, vedrai i seguenti output. Come mostrato, ho il driver GPU NVIDIA ufficiale versione 525.78.01 installato sulla mia macchina Linux Mint 21.

Se non si dispone dei driver GPU NVIDIA ufficiali installati sulla macchina Linux Mint 21 e hai bisogno di assistenza, controlla l'articolo come installare i driver Nvidia su Linux Mint 21.

Installazione delle dipendenze Docker CE da Linux Mint 21

Prima di poter installare Docker CE su Linux Mint 21, dovrai installare i pacchetti di dipendenza Docker CE richiesti su Linux Mint 21.

Innanzitutto, aggiorna la cache del repository del pacchetto APT con il seguente comando:

$ sudo apt update

La cache del repository del pacchetto APT deve essere aggiornata.

Per installare i pacchetti di dipendenza richiesti di Docker CE, eseguire il seguente comando:

$ sudo apt-get Installa certificati CA Curl gnupg LSB-release

Per confermare l'installazione, premere Y e poi premere .

I pacchetti di dipendenza Docker CE richiesti devono essere installati.

Installazione dei tasti GPG Docker CE su Linux Mint 21

In questa sezione, ti mostrerò come installare la chiave gpg del repository pacchetti Docker CE su Linux Mint 21.

Innanzitutto, crea una nuova cartella /etc/apt/keyrings con il seguente comando:

$ sudo mkdir -p/etc/apt/keyrings

Per installare la chiave GPG del repository pacchetto Docker CE su Linux Mint 21, eseguire il seguente comando:

$ CURL -FSSL https: // download.Docker.com/linux/ubuntu/gpg | Sudo GPG - -DEARMOR -O/ETC/APT/KEYRINGS/Docker.gpg

Installazione del repository Docker CE su Linux Mint 21

Per installare il repository del pacchetto Docker CE su Linux Mint 21, eseguire il seguente comando:

$ echo "deb [arch = $ (dpkg --print-architecture) firmato da =/etc/apt/keyrings/docker.gpg] https: // download.Docker.com/linux/ubuntu jammy stabile "| sudo tee ecc/apt/fonti.elenco.D/Docker.Elenco> /dev /null

Aggiorna la cache del repository del pacchetto APT per le modifiche per avere effetto.

$ sudo apt update

Installazione di Docker CE su Linux Mint 21

Per installare l'ultima versione di Docker CE su Linux Mint 21, eseguire il seguente comando:

$ sudo apt-get install docker-ce docker-ce-cli containerd.IO Docker-Opgin-Plugin

Per confermare l'installazione, premere Y e poi premere .

Docker CE e i pacchetti di dipendenza richiesti vengono scaricati. Ci vorrà un po 'di tempo per completare.

Docker CE e i pacchetti di dipendenza richiesti vengono installati. Ci vorrà un po 'di tempo per completare.

Docker CE e i pacchetti di dipendenza richiesti dovrebbero essere installati a questo punto.

Aggiunta dell'utente di accesso Linux Mint 21 al gruppo Docker

Per creare contenitori Docker e gestirli senza utilizzare sudo o accedere come utente root, dovrai aggiungere l'utente di accesso al Docker gruppo.

Per aggiungere l'utente di accesso di Linux Mint 21 al Docker Gruppo, eseguire il seguente comando:

$ sudo usermod -ag docker $ (whoami)

Per avere effetto le modifiche, riavviare il computer con il seguente comando:

$ sudo riavvia

Verifica se Docker CE è stato installato correttamente su Linux Mint 21

Una volta avviato il computer, eseguire il comando seguente per verificare che tu possa accedere a Docker senza privilegi superuser.

$ Docker versione

Se tutto funziona, dovresti vedere i seguenti output.

Come mostrato, sto eseguendo Docker versione 20.10.23 - L'ultima versione di Docker CE al momento della stesura.

Installazione dei tasti GPG per toolkit contenitore NVIDIA su Linux Mint 21

In questa sezione, ti mostrerò come installare la chiave gpg del repository pacchetto toolkit nvidia su Linux Mint 21.

Per installare il tasto GPG del repository pacchetto toolkit Nvidia Container Toolkit su Linux Mint 21, eseguire il seguente comando:

$ CURL -FSSL https: // nvidia.github.IO/libnvidia-contener/gpgkey | Sudo GPG--DEARMOR -O/usr/share/keyrings/nvidia-contener-toolkit-keyring.gpg

Installazione del repository di toolkit del contenitore Nvidia su Linux Mint 21

In questa sezione, ti mostrerò come installare il repository pacchetto toolkit NVIDIA su Linux Mint 21.

Innanzitutto, crea un nuovo file di origine APT Nvidia-contener-toolkit.elenco nel /etc/apt/fonti.elenco.D/ directory come segue:

$ sudo nano/etc/apt/fonti.elenco.d/nvidia-contener-toolkit.elenco

Nel Nvidia-contener-toolkit.elenco file, aggiungi la riga seguente e premi + X seguito da Y E Per salvare il file.

Deb [firmato per =/usr/share/keyrings/nvidia-contener-toolkit-keyring.gpg] https: // nvidia.github.io/libnvidia-container/stable/ubuntu22.04 /$ (arco) /

Per avere effetto sulle modifiche, aggiorna la cache del repository del pacchetto APT con il seguente comando:

$ sudo apt update

Installazione dei driver Nvidia-Docker su Linux Mint 21

Per installare i driver Docker Nvidia su Linux Mint 21, eseguire il seguente comando:

$ sudo apt Installa nvidia-docker2

Per confermare l'installazione, premere Y e poi premere .

I driver Docker Nvidia devono essere installati.

Per avere effetto le modifiche, riavviare il computer con il seguente comando:

$ sudo riavvia

Verifica se la GPU Nvidia è accessibile dai contenitori Docker in Linux Mint 21

In questa sezione, ti mostrerò come creare un contenitore Docker Cuda Nvidia e verificare che il contenitore possa accedere alla GPU NVIDIA dal tuo computer.

Per creare un contenitore Docker Nvidia CUDA 12 basato su Ubuntu 20.04 LTS e eseguire il nvidia-smi comando in esso. Una volta creato per verificare se può accedere alla GPU Nvidia dal tuo computer, eseguire il comando seguente:

$ Docker Run - -RM --gpus All Nvidia/Cuda: 12.0.0-Base-Obuntu20.04 Nvidia-SMI

Docker sta tirando il nvidia/cuda: 12.0.0-Base-Obuntu20.04 Immagine da Docker Hub. Ci vorrà un po 'di tempo per completare.

Una volta creata l'immagine Nvidia Cuda Docker, viene creato un contenitore, il nvidia-smi Il comando verrà eseguito su di esso e stampano l'output sulla console come si può vedere nello screenshot qui sotto.

Come mostrato, il contenitore Docker sta usando il Driver GPU NVIDIA 525.78.01^[1] e il Cuda versione 12.0^[2]. Se vedi output simili, il contenitore Docker può accedere alla GPU NVIDIA del tuo computer.

Se è necessario utilizzare le versioni precedenti di CUDA, controlla l'immagine Nvidia/CUDA su Docker Hub.

Conclusione

Ti ho mostrato come installare il repository dei pacchetti Docker CE su Linux Mint 21, come installare l'ultima versione di Docker CE su Linux Mint 21, come installare il repository pacchetti toolkit NVIDIA su Linux Mint 21 e come installare Il driver Docker Nvidia su Linux Mint 21. Infine, ti ho mostrato come accedere alla GPU NVIDIA del tuo computer da un contenitore Docker.

Riferimenti:

1. https: // docs.nvidia.com/datacenter/cloud-nativo/container-toolkit/install-guide.html
2. https: // hub.Docker.com/r/nvidia/cuda/tag