From 979bb6c3421ee020d0f2ca3b7c51914584b7056d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Luk=C3=A1=C5=A1=20Kucharczyk?= <lukas@kucharczyk.xyz>
Date: Wed, 11 Feb 2026 16:19:04 +0100
Subject: [PATCH] immich-machine-learning: add nvidia version

---
 docker-compose.yml | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/docker-compose.yml b/docker-compose.yml
index e190de4..9294380 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -877,6 +877,31 @@ services:
     environment:
      TZ: ${TZ}
 
+  immich-machine-learning:
+    container_name: immich_machine_learning
+    # For hardware acceleration, add one of -[armnn, cuda, rocm, openvino, rknn] to the image tag.
+    # Example tag: ${IMMICH_VERSION:-release}-cuda
+    image: ghcr.io/immich-app/immich-machine-learning:${IMMICH_VERSION:-release}-cuda
+    # extends: # uncomment this section for hardware acceleration - see https://docs.immich.app/features/ml-hardware-acceleration
+    #   file: hwaccel.ml.yml
+    #   service: cuda # set to one of [armnn, cuda, rocm, openvino, openvino-wsl, rknn] for accelerated inference - use the `-wsl` version for WSL2 where applicable
+    ports:
+      - 3003:3003
+    volumes:
+      - model-cache:/cache
+    restart: always
+    healthcheck:
+      disable: false
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: cdi
+              device_ids:
+                - nvidia.com/gpu=all
+              capabilities:
+                - gpu
+
   slskd:
     image: slskd/slskd:0.24.0
     container_name: slskd
@@ -965,3 +990,4 @@ volumes:
   timetracker-static:
   sockets:
   meilisearch:
+  model-cache: