diff --git a/docker-compose.yml b/docker-compose.yml
index d71034d..f6dd6f0 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -46,6 +46,14 @@ services:
       SESSION_SECRET: ${SESSION_SECRET}
       AUTH_ENABLED: ${AUTH_ENABLED:-false}
       DOCKER_NETWORK: wild-dragon_wild-dragon
+      NODE_IP: ${NODE_IP}
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
     networks:
       - wild-dragon