An Arion NixOS module that requests an NVIDIA GPU

nvidia-test-arion.nix

{ ... }:
{
  virtualisation.arion.projects.nvidia-test.settings.services.nvidia-test = {
    service = {
      image = "nvidia/cuda:12.2.0-base-ubuntu20.04";
      command = "nvidia-smi";
    };
    out.service = {
      deploy.resources.reservations.devices = [
        {
          driver = "nvidia";
          count = 1;
          capabilities = [ "gpu" ];
        }
      ];
    };
  };
}