diff --git a/compose.yaml b/compose.yaml
index 4947379..a03eac6 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -53,10 +53,7 @@ services:
       POSTGRES_PASSWORD: postgres
       POSTGRES_USER: postgres
       MACHINE_LEARNING_MODEL_INTRA_OP_THREADS: 2
-    devices:
-      - /dev/dri:/dev/dri
-    device_cgroup_rules:
-      - 'c 189:* rmw'
+      NVIDIA_VISIBLE_DEVICES: all
     volumes:
       - /dev/bus/usb:/dev/bus/usb
       - /app-config/cache:/cache