Wie kann ich festlegen, nvidia Laufzeit von docker-Komponieren.yml?

Ich bin in der Lage zu laufen ein tensorflow container w/Zugriff auf die GPU über die Befehlszeile w/die folgenden Befehl

$ sudo docker run --runtime=nvidia --rm gcr.io/tensorflow/tensorflow:latest-gpu

Ich würde gerne in der Lage zu sein, diese container von docker-Komponieren. Ist es möglich, an den --runtime Flagge von docker-compose.yml?

InformationsquelleAutor rissem | 2017-11-24
  1. 23

    Sollten Sie Bearbeiten /etc/docker/daemon.json hinzufügen der ersten Ebene Schlüssel "default-runtime": "nvidia" starten docker-daemon (ex. "sudo service docker restart") und dann alle Container, die auf diesem host laufen mit nvidia-Laufzeit.

    Mehr info auf daemon.json hier

    InformationsquelleAutor Gabriel Miretti aka gmiretti
  2. 25

    Derzeit (August 2018), NVIDIA container-runtime für Docker (nvidia-docker2) unterstützt Docker Compose.

    Ja, verwenden Sie format 2.3 und fügen Laufzeit: nvidia Ihre GPU-service. Docker Compose muss in der version 1.19.0 oder höher.

    Beispiel docker-compose.yml:

    version: '2.3'

services:
  nvsmi:
    image: ubuntu:16.04
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    command: nvidia-smi

    Mehr Beispiel von NVIDIA-blog verwendet Docker Compose zu zeigen, wie zu starten, Multi-GPU-Container mit dem NVIDIA-Container-Runtime.

    • Diese option ist nur möglich für docker-verfassen v2 für jetzt. Für mehr info, schauen Sie bitte in den entsprechenden pull-request hier github.com/docker/compose/pull/5405
    • Diese option ist mehr elegant, wie es ermöglicht eine bessere Verpackung mit docker-Komponieren
    InformationsquelleAutor cedrickchee
  3. 3

    Oder besser: mit systemd und vorausgesetzt, der Pfad ist /usr/libexec/oci/hooks.d/nvidia

    Konfigurieren

    mkdir -p /etc/systemd/system/docker.service.d/
cat > /etc/systemd/system/docker.service.d/nvidia-containers.conf <<EOF
[Service]
ExecStart=
ExecStart=/usr/bin/dockerd -D --add-runtime nvidia=/usr/libexec/oci/hooks.d/nvidia --default-runtime=nvidia
EOF

    Neu starten

    systemctl daemon-reload
systemctl restart docker

    Demo

    Nicht angeben müssen --runtime=nvidia da wir default-runtime=nvidia in der Konfiguration Schritt.

    docker run --rm gcr.io/tensorflow/tensorflow:latest-gpu

    Lösung Inspiriert von meinem tutorial über KATA Laufzeit.

    InformationsquelleAutor Abdennour TOUMI
Schreibe einen Kommentar