diff --git a/compose.yaml b/compose.yaml
index f437580..e77764b 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -54,8 +54,10 @@ services:
       POSTGRES_USER: postgres
       MACHINE_LEARNING_MODEL_INTRA_OP_THREADS: 2
     devices:
-      - /dev/dri/card1:/dev/dri/card1
-      - /dev/dri/renderD128:/dev/dri/renderD128
+      - /dev/nvidia0:/dev/nvidia0
+      - /dev/nvidiactl:/dev/nvidiactl
+      - /dev/nvidia-uvm:/dev/nvidia-uvm
+      - /dev/dri:/dev/dri
     device_cgroup_rules:
       - 'c 189:* rmw'
     volumes: