From 8c9a0925611a102bd2dbade1f6e7bda6624789a3 Mon Sep 17 00:00:00 2001 From: Mert <101130780+mertalev@users.noreply.github.com> Date: Thu, 11 Apr 2024 03:39:18 -0400 Subject: [PATCH] docs(ml): update hardware acceleration doc (#8700) * update docs * formatting --- docker/hwaccel.ml.yml | 2 -- docs/docs/features/ml-hardware-acceleration.md | 8 +++----- 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/docker/hwaccel.ml.yml b/docker/hwaccel.ml.yml index 5305271df9..3afc5ceafd 100644 --- a/docker/hwaccel.ml.yml +++ b/docker/hwaccel.ml.yml @@ -27,8 +27,6 @@ services: count: 1 capabilities: - gpu - - compute - - video openvino: device_cgroup_rules: diff --git a/docs/docs/features/ml-hardware-acceleration.md b/docs/docs/features/ml-hardware-acceleration.md index 8f5a889775..1e646a0e7e 100644 --- a/docs/docs/features/ml-hardware-acceleration.md +++ b/docs/docs/features/ml-hardware-acceleration.md @@ -43,7 +43,8 @@ You do not need to redo any machine learning jobs after enabling hardware accele 1. If you do not already have it, download the latest [`hwaccel.ml.yml`][hw-file] file and ensure it's in the same folder as the `docker-compose.yml`. 2. In the `docker-compose.yml` under `immich-machine-learning`, uncomment the `extends` section and change `cpu` to the appropriate backend. -3. Redeploy the `immich-machine-learning` container with these updated settings. +3. Still in `immich-machine-learning`, add one of -[armnn, cuda, openvino] to the `image` section's tag at the end of the line. +4. Redeploy the `immich-machine-learning` container with these updated settings. #### Single Compose File @@ -60,8 +61,6 @@ deploy: count: 1 capabilities: - gpu - - compute - - video ``` You can add this to the `immich-machine-learning` service instead of extending from `hwaccel.ml.yml`: @@ -69,6 +68,7 @@ You can add this to the `immich-machine-learning` service instead of extending f ```yaml immich-machine-learning: container_name: immich_machine_learning + # Note the `-cuda` at the end image: ghcr.io/immich-app/immich-machine-learning:${IMMICH_VERSION:-release}-cuda # Note the lack of an `extends` section deploy: @@ -79,8 +79,6 @@ immich-machine-learning: count: 1 capabilities: - gpu - - compute - - video volumes: - model-cache:/cache env_file: