immich-machine-learning: add nvidia version

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: