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.