diff --git a/compose.yaml b/compose.yaml
index 4e105b4..ea76324 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -21,10 +21,10 @@ services:
       - PUID=1000
       - PGID=1000
       #- NVIDIA_DRIVER_CAPABILITIES=all
-      - NVIDIA_VISIBLE_DEVICES=GPU-4601e5a5-d479-5095-8140-acba49196047
+      #- NVIDIA_VISIBLE_DEVICES=GPU-4601e5a5-d479-5095-8140-acba49196047
     restart: always
     devices:
-      - /dev/nvidia1:/dev/nvidia1
+      - /dev/nvidia0:/dev/nvidia0
       - /dev/nvidiactl:/dev/nvidiactl
       - /dev/nvidia-uvm:/dev/nvidia-uvm
     ports: