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