diff --git a/compose.yaml b/compose.yaml
index 7c4f662..a1809a0 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -28,6 +28,16 @@ services:
     devices:
       - /dev/nvidia0:/dev/nvidia0
       - /dev/nvidiactl:/dev/nvidiactl
+      - /dev/nvidia-uvm:/dev/nvidia-uvm
+      - /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools
+      - /dev/dri/renderD128:/dev/dri/renderD128
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
     ports:
       - "8096:8096"
     networks: