2017-04-12 4 views

Répondre

2

MISE À JOUR: s'il vous plaît vérifier nvidia-docker 2 et son soutien de docker-Compose premier https://github.com/NVIDIA/nvidia-docker/wiki/Frequently-Asked-Questions#do-you-support-docker-compose

(Je voudrais tout d'abord vous suggérons d'ajouter l'étiquette nvidia-docker).

Si vous regardez le code nvidia-docker-compose here, il génère uniquement un docker-fichier spécifique pour docker-composer après une requête de la configuration de nvidia sur localhost: 3476.

Vous pouvez également créer manuellement ce fichier Docker-Composer car il s'avère assez simple, suivez cet exemple, remplacez 375.66 par votre version de pilote nvidia et mettez autant de lignes/dev/nvidia [n] que vous avez Les cartes graphiques (n'a pas essayé de mettre des services séparés mais GPUs y aller!):

services: 
    exampleservice0: 
    devices: 
    - /dev/nvidia0 
    - /dev/nvidia1 
    - /dev/nvidiactl 
    - /dev/nvidia-uvm 
    - /dev/nvidia-uvm-tools 
    environment: 
    - EXAMPLE_ENV_VARIABLE=example 
    image: company/image 
    volumes: 
    - ./disk:/disk 
    - nvidia_driver_375.66:/usr/local/nvidia:ro 
version: '2' 
volumes: 
    media: null 
    nvidia_driver_375.66: 
    external: true 

Ensuite, il suffit d'exécuter ce fichier docker-Compose fait à la main avec un classique commande docker-Compose.

Peut-être que vous pouvez ensuite composer avec Dockers non nvidia en sautant les trucs spécifiques nvidia dans les autres services.