Configura Docker per utilizzare il contenitore compatibile con GPU Nvidia [Ubuntu 18.04]
Docker ha il supporto nativo per "Container Acceleration con GPU Nvidia" dalla versione 19.03.Questa funzionalità è disponibile utilizzando NVIDIA Container Toolkit.Qui, effettua le impostazioni iniziali necessarie.
ambiente
Sistema operativo: Ubuntu 18.04 LTS
GPU: GTX1080
- 1. Passaggio XNUMX: verifica i prerequisiti
- 2. Procedura di installazione di Nvidia docker 2 e conferma dei prerequisiti [Ubuntu 18.04]
- 3. Passaggio XNUMX: installa Docker
- 4. Installa l'ultima versione di Docker CE su Ubuntu 18.04 [conforme al documento]
- 5. Passaggio XNUMX: installa Nvidia GPU Driver
- 6. Due modi per installare il driver della GPU NVIDIA [Ubuntu 18.04]
- 7. Passaggio XNUMX: rendere disponibile il contenitore GPU in Docker
- 8. Verificare il funzionamento
Passaggio XNUMX: verifica i prerequisiti
Prima di utilizzare "l'accelerazione dei container tramite GPU" di Docker, controlliamo l'ambiente.I contenitori GPU non possono essere utilizzati in ambienti troppo vecchi.
- GNU / Linux x86_64 (versione kernel> 3.10) ← Ubuntu 18.04 quindi OK
- GPU NVIDIA con architettura> Fermi (2.1) ← Se la GPU è per desktop, la GTX 600 o successiva è OK
- Versione driver NVIDIA ≧ 361.93 ← Vedere il passaggio XNUMX di seguito
Si prega di consultare la pagina seguente per i dettagli.
Passaggio XNUMX: installa Docker
Se non hai ancora installato Docker, segui i passaggi seguenti per installarlo.
Passaggio XNUMX: installa Nvidia GPU Driver
Il driver Nvidia è necessario per eseguire il contenitore GPU (contenitore CUDA).
La versione del driver Nvidia dovrebbe essere maggiore della versione di CUDA che desideri utilizzare (ad esempio, tensorflow richiede esattamente la versione CUDA a seconda della versione che utilizzi).
Finché si installa correttamente il driver Nvidia, non è necessario installare separatamente CUDA toolkit o NVIDIA Container Toolkit.
Il metodo di installazione di Nvidia Driver è riassunto di seguito.
Passaggio XNUMX: rendere disponibile il contenitore GPU in Docker
Immettere quanto segue dal terminale.
Riferimento ufficiale: https://github.com/NVIDIA/nvidia-docker
#Add distribuzione del repository del pacchetto = $ (. / Etc / os-release; echo $ ID $ VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add- curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit sudo systemctl riavvia docker
Verificare il funzionamento
# Testato con l'ultima versione ufficiale della finestra mobile di immagini CUDA eseguita --gpus all nvidia / cuda: 9.0-base nvidia-smi
Se tutto va bene, vedrai questo display.
Per utilizzare il contenitore GPU d'ora in poi, aggiungi l'opzione "–gpus" a "docker run".
Se hai appena installato docker, potrebbe non funzionare a meno che tu non aggiunga sudo prima di docker run.
[Ubuntu] Impostazioni e precauzioni per l'esecuzione di docker senza il comando sudo
Queste sono le impostazioni iniziali richieste per utilizzare il contenitore compatibile con la GPU di Docker su Ubuntu.
Libro di consultazione