margipose/docker-compose.yml.example at master · anibali/margipose · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
version: '3.7'
services:
  main:
    build: .
    volumes:
      # Volume mounts for the datasets.
      # Edit to suit your environment. Remove entries for datasets that you don't intend to
      # use.
      - /host/path/to/h36m:/datasets/h36m:ro
      - /host/path/to/mpi3d:/datasets/mpi3d:ro
      - /host/path/to/mpii:/datasets/mpii:rw
      # Other volume mounts.
      - .:/app
      - ~/.cache/torch/checkpoints:/home/user/.cache/torch/checkpoints:rw
      - /etc/localtime:/etc/localtime:ro
      - /tmp/.X11-unix:/tmp/.X11-unix:rw
      - /etc/hostname:/etc/hostname:ro
    environment:
      - CUDA_DEVICE_ORDER=PCI_BUS_ID
      - DISPLAY
      - PYTHONIOENCODING=utf_8
#    env_file:
#      - showoff/showoff-client.env
    init: true
    ipc: host
    network_mode: host
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]