public
/
immich-ml-container
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use driver name in device name (and fix debug)

This commit is contained in:
niten 2024-05-22 10:44:02 -07:00
parent 713b1c42c2
commit 0fd4340177
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
immich-ml-container.nix
View File

@ -13,15 +13,12 @@ let
services.immich-ml = {
image =
"ghcr.io/immich-app/immich-machine-learning:${cfg.immich-version}-cuda";
deploy.resources.reservations.devices = [{
driver = "nvidia";
count = 1;
capabilities = [ "gpu" "compute" "video" ];
}];
deploy.resources.reservations.devices =
[{ capabilities = [ "nvidia-gpu" "nvidia-compute" "nvidia-video" ]; }];
ports = [ "${toString cfg.port}:3003" ];
restart = "always";
volumes = [ "${cfg.state-directory}:/cache" ];
env = mkIf cfg.debug { IMMICH_LOG_LEVEL = "debug"; };
environment = mkIf cfg.debug { IMMICH_LOG_LEVEL = "debug"; };
};
});