diff --git a/docker-compose.yml b/docker-compose.yml
index 3ca5072..c3a88bb 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -4,4 +4,10 @@ services:
   subsai-webui:
     build:
       context: .
-      dockerfile: Dockerfile
\ No newline at end of file
+      dockerfile: Dockerfile
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              capabilities: [ gpu ]
\ No newline at end of file