Skip to content

Nvidia-docker-compose

Simple wrapper for docker-compose to use GPU enabled docker under nvidia-docker

Installing

To install the script, simply run: 

$ pip install nvidia-docker-compose

Example 

version: '3.3'
services:
  core:
    image: your/proj-core:1.1.1-ubuntu1804-gpu
    devices:
      - /dev/nvidia0
      - /dev/nvidiactl
      - /dev/nvidia-uvm
    restart: always
    network_mode: host
    volumes:
      - /home/your/core/node:/usr/local/c2core/storage/node
      - /home/your/core/python:/usr/local/c2core/storage/python
      - /home/your/core/workspace:/usr/local/c2core/storage/workspace
    command:
      - --verbose
      - --sync
      - node
    entrypoint:
      - /usr/local/c2core/bin/c2node

See also

Favorite site