diff --git a/compose.yaml b/compose.yaml
index 0f2073d..71c2882 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -16,10 +16,13 @@ services:
       - PUID=1000
       - PGID=1000
       - OutputLogsToDocker=1
+      - 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: