diff --git a/compose.yaml b/compose.yaml
index 0b0c89b..09d1717 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -16,8 +16,13 @@ services:
       - PGID=1000
       - NVIDIA_DRIVER_CAPABILITIES=all
       - NVIDIA_VISIBLE_DEVICES=all
-    devices:
-      - /dev/dri/:/dev/dri/
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
 
 volumes:
   usenet: