UNRAID 使用 CUDA 加速 Immich 相似照片去重

1
name: immich
2

3
services:
4
  immich-server:
5
    container_name: immich_server
6
    image: ghcr.io/immich-app/immich-server:release
7
    # extends:
8
    #   file: hwaccel.transcoding.yml
9
    #   service: cpu # set to one of [nvenc, quicksync, rkmpp, vaapi, vaapi-wsl] for accelerated transcoding
10
    volumes:
11
      - /mnt/user/nas/Photos/libraries:/usr/src/app/libraries
12
      - /mnt/user/nas/Photos/immich:/usr/src/app/upload
13
      - /etc/localtime:/etc/localtime:ro
14
    environment:
15
      - MACHINE_LEARNING_GPU_ACCELERATION=cuda
16
      - NVIDIA_VISIBLE_DEVICES=all
17
    ports:
18
      - '2283:2283'
19
    depends_on:
20
      - redis
21
      - database
22
    restart: always
23
    healthcheck:
24
      disable: false
25
    runtime: nvidia
26

27
  immich-machine-learning:
28
    container_name: immich_machine_learning
29
    # For hardware acceleration, add one of -[armnn, cuda, rocm, openvino, rknn] to the image tag.
30
    # Example tag: ${IMMICH_VERSION:-release}-cuda
31
    image: ghcr.io/immich-app/immich-machine-learning:${IMMICH_VERSION:-release}-cuda
32
    deploy:
33
      resources:
34
        reservations:
35
          devices:
36
            - driver: nvidia
37
              count: 1
38
              capabilities:
39
                - gpu
40
    volumes:
41
      - /mnt/user/nas/Photos/model-cache:/cache
42
    restart: always
43
    healthcheck:
44
      disable: false
45

46
  redis:
47
    container_name: immich_redis
48
    image: docker.io/valkey/valkey:8-bookworm@sha256:fec42f399876eb6faf9e008570597741c87ff7662a54185593e74b09ce83d177
49
    healthcheck:
50
      test: redis-cli ping || exit 1
51
    restart: always
52

53
  database:
54
    container_name: immich_postgres
55
    image: ghcr.io/immich-app/postgres:14-vectorchord0.4.3-pgvectors0.2.0
56
    environment:
57
      POSTGRES_PASSWORD: postgres
58
      POSTGRES_USER: postgres
59
      POSTGRES_DB: immich
60
      POSTGRES_INITDB_ARGS: '--data-checksums'
61
      # Uncomment the DB_STORAGE_TYPE: 'HDD' var if your database isn't stored on SSDs
62
      DB_STORAGE_TYPE: 'HDD'
63
    volumes:
64
      - /mnt/user/nas/Photos/postgres:/var/lib/postgresql/data
65
    restart: always
66
    ports:
67
      - "5432:5432"

1
name: immich-mediakit
2
services:
3
  immich-mediakit:
4
    container_name: immich-mediakit
5
    hostname: immich-mediakit
6
    image: razgrizhsu/immich-mediakit:latest-cuda
7
    restart: unless-stopped
8
    ports:
9
      - 8086:8086
10
    volumes:
11
      - /mnt/user/nas/immich-mediakit:/app/data                  # mediakit data
12
      - /mnt/user/nas/immich-mediakit/cache:/root/.cache/torch/  # model cache
13
      - /mnt/user/nas/Photos/immich:/immich:ro
14
    env_file:
15
      - .env
16
    environment:
17
      - DASH_DEBUG=false
18
      - IMMICH_PATH=/immich
19
    # GPU Support (Linux + NVIDIA GPU only)
20
    # To enable NVIDIA GPU acceleration:
21
    # 1. Change image tag to razgrizhsu/immich-mediakit:latest-cuda
22
    # 2. Uncomment the deploy configuration below
23
    deploy:
24
      resources:
25
        reservations:
26
          devices:
27
            - driver: nvidia
28
              count: 1
29
              capabilities: [gpu]
30

31
    depends_on:
32
      - qdrant
33
    healthcheck:
34
      test: sh -c "curl -f http://localhost:8086/ || exit 1"
35
      interval: 30s
36
      timeout: 10s
37
      retries: 3
38
      start_period: 40s
39
    networks:
40
      - immich-mediakit
41

42
  qdrant:
43
    container_name: immich-mediakit-qdrant
44
    image: qdrant/qdrant:dev
45
    restart: unless-stopped
46
    ports:
47
      - 6333:6333
48
    volumes:
49
      - qdrant:/qdrant/storage
50
      #- /mnt/user/nas/immich-mediakit/qdrant:/qdrant/storage
51
    networks:
52
      - immich-mediakit
53

54
networks:
55
  immich-mediakit:
56
    driver: bridge
57

58
volumes:
59
  qdrant:

1
# MediaKit settings
2
MKIT_PORT=8086
3
MKIT_DATA=./data
4

5

6
# Immich library connection (UPLOAD_LOCATION env variable of Immich if you're using Immich with docker compose)
7
# This folder should contain the 'thumbs', 'library' subfolder from your Immich installation.
8
IMMICH_PATH=/immich
9

10

11
# PostgreSQL connection to Immich
12
PSQL_HOST=192.168.101.236
13
PSQL_PORT=5432
14
PSQL_DB=immich
15
PSQL_USER=postgres
16
PSQL_PASS=postgres
17

18
# For different-host setup:
19
# You must change PSQL_HOST to the IP address of your Immich server
20
#
21
# Example:
22
# Immich running on 192.168.0.100
23
# MediaKit running on 192.168.0.200
24
# Change localhost above to 192.168.0.100

1
Error validating vector storage: Unexpected Response: 500 (Internal Server Error)
2
Raw response content:
3
b'{"status":{"error":"Service internal error: 1 of 1 read operations failed:\\n  Service internal error: task 3802 panicked with message \\"called `Result::unwrap()` on an `Err` value: OutputTooSmall { ...'
4
07:50:16.559|ERRO| Error saving vector for asset 902: Unexpected Response: 500 (Internal Server Error)
5
Raw response content:
6
b'{"status":{"error":"Service internal error: 1 of 1 read operations failed:\\n  Service internal error: task