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

Switch to using oci-containers

This commit is contained in:
niten 2024-05-22 18:17:58 -07:00
parent f6871bbc20
commit 238ef10160
1 changed files with 11 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
immich-ml-container.nix
View File

@ -1,26 +1,7 @@
{ config, lib, pkgs, ... }: { config, lib, pkgs, ... }:
with lib; with lib;
let let cfg = config.services.immichMlContainer;
cfg = config.services.immichMlContainer;
immichMlConfigYaml = pkgs.writeText "immich-ml-compose.yml" (builtins.toJSON {
version = "3.8";
name = "immich-machine-learning";
networks.default.name = "immich-ml";
volumes = { };
services.immich-ml = {
image =
"ghcr.io/immich-app/immich-machine-learning:${cfg.immich-version}-cuda";
deploy.resources.reservations.devices =
[{ capabilities = [ "nvidia-gpu" "nvidia-compute" "nvidia-video" ]; }];
ports = [ "${toString cfg.port}:3003" ];
restart = "always";
volumes = [ "${cfg.state-directory}:/cache" ];
environment = { IMMICH_LOG_LEVEL = "debug"; };
};
});
in { in {
options.services.immichMlContainer = with types; { options.services.immichMlContainer = with types; {
@ -53,19 +34,17 @@ in {
}; };
config = mkIf cfg.enable { config = mkIf cfg.enable {
systemd = { systemd.tmpfiles.rules = [ "d ${cfg.state-directory} 0750 root root - -" ];
services.immich-machine-learning = {
after = [ "network-online.target" ];
before = [ "nginx.service" ];
path = with pkgs; [ docker-compose nvidia-podman coreutils ];
serviceConfig = {
ExecStart = pkgs.writeShellScript "immich-machine-learning" ''
docker-compose -f ${immichMlConfigYaml} up
'';
};
};
tmpfiles.rules = [ "d ${cfg.state-directory} 0750 root root - -" ]; virtualisations.oci-containers.containers = {
immich-machine-learning = {
autoStart = true;
image =
"ghcr.io/immich-app/immich-machine-learning:${cfg.immich-version}-cuda";
volumes = [ "${cfg.state-directory}:/cache" ];
ports = [ "${cfg.port}:3003" ];
extraOptions = [ "--gpus=all" ];
};
}; };
services.nginx = { services.nginx = {