Richten Sie Docker für die Verwendung des Nvidia GPU-kompatiblen Containers ein [Ubuntu 18.04]
Ab Version 19.03 unterstützt Docker nativ „Containerbeschleunigung mit Nvidia-GPUs“.Diese Funktionalität ist mit dem NVIDIA Container Toolkit verfügbar.Hier werden wir die notwendigen Grundeinstellungen dafür vornehmen.
Umwelt
Betriebssystem: Ubuntu 18.04 LTS
GPU: GTX1080
- 1. Schritt XNUMX: Voraussetzungen prüfen
- 2. Installationsverfahren für Nvidia Docker 2 und Bestätigung der Voraussetzungen [Ubuntu 18.04]
- 3. Schritt XNUMX: Docker installieren
- 4. Installieren Sie die neueste Version von Docker CE unter Ubuntu 18.04 [Dokumentkonform]
- 5. Schritt XNUMX: Installieren Sie die Nvidia GPU-Treiber
- 6. Zwei Möglichkeiten zur Installation des NVIDIA GPU-Treibers [Ubuntu 18.04]
- 7. Schritt XNUMX: Aktivieren Sie Docker für die Verwendung von GPU-Containern
- 8. Funktion prüfen
Schritt XNUMX: Voraussetzungen prüfen
Bevor wir Dockers „Containerbeschleunigung mit GPU“ verwenden, überprüfen wir die Umgebung.GPU-Container können nicht in zu alten Umgebungen verwendet werden.
- GNU/Linux x86_64 (Kernel-Version > 3.10) ← Ubuntu 18.04 ist in Ordnung
- NVIDIA-GPU mit Architektur > Fermi (2.1) ← Wenn die GPU für den Desktop geeignet ist, ist GTX600 oder höher in Ordnung
- NVIDIA-Treiberversion ≧ 361.93 ← Siehe Schritt XNUMX unten
Weitere Informationen finden Sie auf der folgenden Seite.
Schritt XNUMX: Docker installieren
Wenn Sie Docker noch nicht installiert haben, führen Sie die folgenden Schritte aus, um es zu installieren.
Schritt XNUMX: Installieren Sie die Nvidia GPU-Treiber
Für die Ausführung von GPU-Containern (CUDA-Containern) sind Nvidia-Treiber erforderlich.
Die Nvidia-Treiberversion sollte mehr als kompatibel mit der CUDA-Version sein, die Sie verwenden möchten (z. B. Tensorflow usw. erfordern unterschiedliche genaue CUDA-Versionen, je nachdem, welche Version Sie verwenden).
Solange Sie die Nvidia-Treiber ordnungsgemäß installieren, müssen Sie das CUDA Toolkit oder das NVIDIA Container Toolkit nicht separat installieren.
Hier erfahren Sie, wie Sie die Nvidia-Treiber installieren.
Schritt XNUMX: Aktivieren Sie Docker für die Verwendung von GPU-Containern
Geben Sie an Ihrem Terminal Folgendes ein:
Offizielle Referenz: https://github.com/NVIDIA/nvidia-docker
# Paket-Repository hinzufügen distribution=$(. /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 restart docker
Funktion prüfen
# Testen Sie mit dem neuesten offiziellen CUDA-Image-Docker, führen Sie --gpus all nvidia/cuda:9.0-base nvidia-smi aus
Wenn alles gut geht, werden Sie so etwas sehen.
Um von nun an den GPU-Container zu verwenden, fügen Sie die Option „-gpus“ zu „docker run“ hinzu.
Wenn Sie Docker gerade erst installiert haben, wird es möglicherweise nicht ausgeführt, es sei denn, Sie fügen sudo vor der Docker-Ausführung hinzu.
[Ubuntu] Einstellungen und Vorsichtsmaßnahmen zum Ausführen von Docker ohne den Befehl sudo
Oben sind die anfänglichen Einstellungen aufgeführt, die für die Verwendung von Docker-GPU-fähigen Containern unter Ubuntu erforderlich sind.
Nachschlagewerk