My docker-compose.yml file :

version: '2'
services:
  zl:
    image: zl/caffe-torch-gpu:12.27
    ports:
      - "8801:8888"
      - "6001:6008"
    devices:
      - /dev/nvidia0
    volumes:
      - ~/dl-data:/root/dl-data

After nvidia-docker-compose up -d容器启动了，但很快就退出了。

但是当我启动一个容器时nvidia-docker方式，效果很好。

nvidia-docker run -itd -p 6008:6006 -p 8808:8888 -v `pwd`:/root/dl-data --name zl_test 

您不必使用 nvidia-docker-compose。 通过正确配置 nvidia-docker 插件，您就可以使用 docker-compose！

Via the Nvidia docker git 存储库 https://github.com/NVIDIA/nvidia-docker/issues/39: （可以确认它对我有用）

Step 1:

找出 nvidia 驱动程序版本（这很重要）。 跑步：

 nvidia-smi

output:

+------------------------------------------------ --------------+

NVIDIA-SMI 367.57 驱动程序版本：367.57

|--------------------------------+--------+-------- --------------+

Step 2:

创建使用 nvidia-docker 插件的 docker 卷必须在 compose 之外完成，因为 compose 如果创建它，将会破坏卷名称。

docker volume create --name=nvidia_driver_367.57 -d nvidia-docker

Step 3

在 docker-compose.yml 文件中：

version: '2'
    volumes:
      nvidia_driver_367.57: # same name as one created above
        external: true  #this will use the volume we created above

    services:
      cuda:
        command: nvidia-smi
        devices:  #this is required
        - /dev/nvidiactl
        - /dev/nvidia-uvm
        - /dev/nvidia0 #in general: /dev/nvidia# where # depends on which gpu card is wanted to be used
        image: nvidia/cuda
        volumes:
        - nvidia_driver_367.57:/usr/local/nvidia/:ro