wyoming-addons/docker-compose.gpu.yml at a8f07f2d66070a4791f7595be68152ce45e050aa · rhasspy/wyoming-addons · 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
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
### YAML Anchors ###
x-gpu: &gpu
  build:
    args:
      - BASE=nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04
  deploy:
    resources:
      reservations:
        devices:
          - driver: nvidia
            count: all
            capabilities: ["compute", "utility", "graphics"]

####
services:
  wyoming-piper:
    extends:
      file: docker-compose.base.yml
      service: wyoming-piper
    <<: [ *gpu ]
    build:
      dockerfile: GPU.Dockerfile
      args:
        - EXTRA_DEPENDENCIES=onnxruntime-gpu
        - RUN_SCRIPT=run-gpu.sh
    volumes:
      - ./piper/__main__.py:/opt/venv/lib/python3.11/site-packages/wyoming_piper/__main__.py
      - ./piper/process.py:/opt/venv/lib/python3.11/site-packages/wyoming_piper/process.py

  wyoming-whisper:
    extends:
      file: docker-compose.base.yml
      service: wyoming-whisper
    <<: [ *gpu ]
    command: [ "--model", "tiny-int8", "--language", "en", "--device", "cuda" ]

#  wyoming-whispercpp:
#    extends:
#      file: docker-compose.base.yml
#      service: wyoming-whispercpp
#    <<: [ *gpu ]
#    command: [ "--model", "tiny-int8", "--language", "en", "--device", "cuda" ]

  wyoming-openwakeword:
    extends:
      file: docker-compose.base.yml
      service: wyoming-openwakeword
    <<: [ *gpu ]

#  wyoming-porcupine:
#    extends:
#      file: docker-compose.base.yml
#      service: wyoming-porcupine
#    <<: [ *gpu ]

#  wyoming-snowboy:
#    extends:
#      file: docker-compose.base.yml
#      service: wyoming-snowboy
#    <<: [ *gpu ]

#  speech-to-phrase:
#    extends:
#      file: docker-compose.base.yml
#      service: speech-to-phrase
#    <<: [ *gpu ]

#  wyoming-vosk:
#    extends:
#      file: docker-compose.base.yml
#      service: wyoming-vosk
#    <<: [ *gpu ]

#  wyoming-microwakeword:
#    extends:
#      file: docker-compose.base.yml
#      service: wyoming-microwakeword
#    <<: [ *gpu ]

#  wyoming-rhasspy-speech:
#    extends:
#      file: docker-compose.base.yml
#      service: wyoming-rhasspy-speech
#    <<: [ *gpu ]