File size: 1,952 Bytes
50f034f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

version: "3.5"
services:
  tensorrt:
    build:
      context: .
      dockerfile: ./deployment_build/dockerfile
      args:
        GITHUB_TOKEN: ${GITHUB_TOKEN}
    profiles:
      - tensorrt
    volumes:
      - ./:/u/home/src
      - ~/.config/gcloud:/root/.config/gcloud
    deploy:
      resources:
        limits:
          # match production limits
          cpus: '7'
          memory: 25000M
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    working_dir: /u/home/src
    entrypoint:
      - python
      - -m
      - wav2wav.converter
  base:
    build:
      context: .
      dockerfile: ./Dockerfile
      args:
        GITHUB_TOKEN: ${GITHUB_TOKEN}
    volumes:
      - .:/u/home/src
      - ~/.wav2wav:/u/home/.wav2wav
      - ${PATH_TO_DATA}:/data
      - ${PATH_TO_RUNS}:/runs
      - ~/.config/gcloud:/u/home/.config/gcloud
      - ~/.zsh_history:/u/home/.zsh_history
    environment:
      - GITHUB_TOKEN
      - DISCOURSE_API_USERNAME
      - DISCOURSE_SERVER
      - DISCOURSE_API_KEY
      - HOST_USER_ID
      - HOST_USER_GID
      - JUPYTER_TOKEN
      - PATH_TO_DATA=/data
      - PATH_TO_RUNS=/runs
      - TENSORBOARD_PATH
      - MPLCONFIGDIR=/u/home/.mplconfig
    shm_size: 32G
    working_dir: /u/home/src
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              capabilities: [gpu]
  dev:
    extends: base
    profiles:
      - interactive
    stdin_open: true
    tty: true
  jupyter:
    extends: base
    ports:
      - ${JUPYTER_PORT}:8888
    entrypoint:
      - /bin/bash
      - /entry_script_jupyter.sh
  tensorboard:
    extends: base
    ports:
      - ${TENSORBOARD_PORT}:6006
    entrypoint:
      - /bin/bash
      - /entry_script_tensorboard.sh
  gradio:
    extends: base
    ports:
      - 7860:7860
    entrypoint:
      - /bin/bash
      - /entry_script_gradio.sh