diff --git a/compose.yaml b/compose.yaml
index cfe02fb..f12ccc7 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -20,10 +20,13 @@ services:
       - TZ=Europe/Berlin
       - PUID=1000
       - PGID=1000
+      - NVIDIA_DRIVER_CAPABILITIES=all
+      - NVIDIA_VISIBLE_DEVICES=all
     restart: always
     devices:
-      - /dev/dri/card0:/dev/dri/card0
-      - /dev/dri/renderD128:/dev/dri/renderD128
+      - /dev/nvidia0:/dev/nvidia0
+      - /dev/nvidiactl:/dev/nvidiactl
+      - /dev/nvidia-uvm:/dev/nvidia-uvm
     ports:
       - "8096:8096"
     networks: