{ pkgs, config, ... }:

let
  ollamaPort = "4827";
in
{
  services.ollama.listenAddress = "0.0.0.0:${ollamaPort}";
  services.ollama.environmentVariables = {
    "CUDA_VISIBLE_DEVICES" = "GPU-cf2321f0-d34b-ec9b-31e9-8c0c69e1444a";
    OLLAMA_LLM_LIBRARY = "gpu";
  };
  services.ollama.enable = true;
  services.ollama.acceleration = "cuda";


}