Immich and CUDA-accelerated machine learning

Hi,

I’m trying to set up CUDA-acceleration for the machine learning service of Immich.

This is running on a computer with an Nvidia T500, which should be supported by Immich.

GPU is set up like this (which is more or less copy-pasted from the wiki):

{ config, ... }:
{
  allowedUnfree = [ "nvidia-x11" "nvidia-persistenced" ];

  # Enable OpenGL
  hardware.graphics = {
    enable = true;
  };
  services.xserver.videoDrivers = [ "nvidia" ];

  hardware.nvidia = {
    nvidiaPersistenced = false;
    # Modesetting is required.
    modesetting.enable = true;

    # Nvidia power management. Experimental, and can cause sleep/suspend to fail.
    # Enable this if you have graphical corruption issues or application crashes after waking
    # up from sleep. This fixes it by saving the entire VRAM memory to /tmp/ instead 
    # of just the bare essentials.
    powerManagement.enable = false;

    # Fine-grained power management. Turns off GPU when not in use.
    # Experimental and only works on modern Nvidia GPUs (Turing or newer).
    powerManagement.finegrained = false;

    # Use the NVidia open source kernel module (not to be confused with the
    # independent third-party "nouveau" open source driver).
    # Support is limited to the Turing and later architectures. Full list of 
    # supported GPUs is at: 
    # https://github.com/NVIDIA/open-gpu-kernel-modules#compatible-gpus 
    # Only available from driver 515.43.04+
    # Currently alpha-quality/buggy, so false is currently the recommended setting.
    open = false;

    # Enable the Nvidia settings menu,
    # accessible via `nvidia-settings`.
    nvidiaSettings = false;

    # Optionally, you may need to select the appropriate driver version for your specific GPU.
    package = config.boot.kernelPackages.nvidiaPackages.stable;

    prime = {
      offload = {
        enable = true;
        enableOffloadCmd = true;
      };
      intelBusId = "PCI:0:2:0";
      nvidiaBusId = "PCI:1:0:0";
    };
  };

  hardware.nvidia-container-toolkit.enable = true;
}

Immich is configured like this:

  services.immich = {
    enable = true;
    openFirewall = true;
    host = "0.0.0.0";
}

My problem is that the GPU doesn’t seem to be recognised. I test this by uploading a new image to Immich and looking at nvtop. I expect the “smart search” machine learning job to cause some load on the GPU, but never see some.

Already tried to set
services.immich.machine-learning.environment.DEVICE to cuda or nvidia and users.users.immich.extraGroups = ["video" "render"];, but didn’t succeed.

What am I missing?

1 Like

Oh, it seems the package is not built with GPU support. Compare the dependencies in the package to the original project.toml.
Can someone confirm that my interpretation is correct?