nixpkgs/nixos/tests/ollama-cuda.nix

{ lib, pkgs, ... }:
{
  name = "ollama-cuda";
  meta.maintainers = [ ];

  nodes.cuda =
    { ... }:
    {
      services.ollama.enable = true;
      services.ollama.package = pkgs.ollama-cuda;
    };

  testScript = ''
    cuda.wait_for_unit("multi-user.target")
    cuda.wait_for_open_port(11434)
  '';
}