diff --git a/compose.yaml b/compose.yaml
index 18fe60f..8491c75 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -24,10 +24,9 @@ services:
       - NVIDIA_VISIBLE_DEVICES=all
     restart: always
     devices:
-      - /dev/nvidia0:/dev/nvidia0
+      - /dev/nvidia1:/dev/nvidia1
       - /dev/nvidiactl:/dev/nvidiactl
       - /dev/nvidia-uvm:/dev/nvidia-uvm
-      - /dev/dri:/dev/dri
     ports:
       - "8096:8096"
     networks: