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: