diff --git a/compose.yaml b/compose.yaml
index 6d2d80d..4758c28 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -43,7 +43,7 @@ services:
 
   immich-machine-learning:
     container_name: immich_machine_learning
-    image: ghcr.io/immich-app/immich-machine-learning:release-openvino
+    image: ghcr.io/immich-app/immich-machine-learning:release-cuda
     restart: always
     healthcheck:
       disable: false
@@ -55,7 +55,7 @@ services:
       MACHINE_LEARNING_MODEL_INTRA_OP_THREADS: 2
     devices:
       - /dev/dri/card1:/dev/dri/card1
-      - /dev/dri/renderD129:/dev/dri/renderD129
+      - /dev/dri/renderD128:/dev/dri/renderD128
     device_cgroup_rules:
       - 'c 189:* rmw'
     volumes:
@@ -90,5 +90,3 @@ services:
       - /etc/localtime:/etc/localtime:ro
     networks:
       - immich_network
-    devices:
-      - /dev/dri/:/dev/dri/