nixos/tests/ollama-cuda.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

{ 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)
  '';
}