diff --git a/docker-compose.yml b/docker-compose.yml
index 852d4c0..15add31 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -21,4 +21,12 @@ services:
     group_add:
       - '993'
     devices:
-      - /dev/dri/renderD129:/dev/dri/renderD129
\ No newline at end of file
+      - /dev/dri:/dev/dri
+    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+              - driver: nvidia
+                count: all
+                capabilities: [ gpu ]
\ No newline at end of file