diff --git a/docker-compose.yml b/docker-compose.yml
index 7c25ee6..9300b70 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -20,5 +20,12 @@ services:
     volumes:
       - /var/lib/docker/volumles/vdup/config:/config:rw
       - /network:/storage:rw
-    devices:
-      - /dev/dri:/dev/dri
\ No newline at end of file
+    # devices:
+    #   - /dev/dri:/dev/dri
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [ gpu ]
+
+    runtime: nvidia
\ No newline at end of file