diff --git a/compose.yaml b/compose.yaml
index e289c26..68d5291 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -16,8 +16,8 @@ services:
       - PUID=1000
       - PGID=1000
       - OutputLogsToDocker=1
-      - NVIDIA_DRIVER_CAPABILITIES=all
-      - NVIDIA_VISIBLE_DEVICES=all
+      #- NVIDIA_DRIVER_CAPABILITIES=all
+      - NVIDIA_VISIBLE_DEVICES=GPU-ce2493b8-b88d-caca-e87b-0f97b3735569
     restart: always
     devices:
       - /dev/nvidia0:/dev/nvidia0