diff options
Diffstat (limited to 'docker-compose.yaml')
| -rw-r--r-- | docker-compose.yaml | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/docker-compose.yaml b/docker-compose.yaml index af14ad6..f040bc3 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -27,3 +27,10 @@ services: - 8000:8000 volumes: - ./deep_phonemizer/assets/model.pt:/app/assets/model.pt + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: [gpu]
\ No newline at end of file |