使用docker-compose中的nvidia-docker

我想用docker-compose运行2个docker镜像。

一个图像应该与nvidia-docker运行,另一个运行docker。

我已经看到这个post使用nvidia-docker-compose启动一个容器,但很快退出,但这不适合我(即使只运行一个图像)…

任何想法将是伟大的。

(我首先build议添加nvidia-docker标签)。

如果您查看nvidia-docker-compose代码,则只会在localhost:3476上的nvidiaconfiguration查询后,为docker-compose生成一个特定的docker-file。

你也可以手工制作这个docker-compose文件,因为它们变得非常简单,按照这个例子,用你的nvidia驱动程序版本replace375.66,并把你的graphics卡放到/ dev / nvidia [n]行中不要尝试把服务放在单独的GPU上,但要去做!):

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 

然后用一个经典的docker-compose命令运行这个手工制作的docker-compose文件。

也许你可以通过跳过其他服务中的nvidia特定的东西来与非nvidia的dockers合作。