Install ollama (support for cuda)

hosts/pro-laptop/configuration.nix

@@ -47,5 +47,9 @@
       printing.enable = true;
       waydroid.enable = true;
     };
+
+    server = {
+      ollama.enable = true;
+    };
   };
 }

modules/system/server/default.nix

@@ -3,6 +3,7 @@
 {
   imports = [
     ./docker
+    ./ollama
     ./openssh
   ];
 }

modules/system/server/ollama/default.nix

@@ -0,0 +1,21 @@
+{ config, pkgs, lib, ... }:
+
+with lib;
+let
+  cfg = config.modules.system.server.ollama;
+  nvidiaEnabled = config.modules.system.hardware.nvidia.enable;
+in
+{
+  options.modules.system.server.ollama = {
+    enable = mkEnableOption ''
+      Enable ollama with my custom configurations
+    '';
+  };
+  config = mkIf cfg.enable {
+    services.ollama = {
+      enable = true;
+
+      acceleration = if config.modules.system.hardware.nvidia.enable then "cuda" else null;
+    };
+  };
+}