3 files changed, 29 insertions, 0 deletions
diff --git a/nixos/opt/default.nix b/nixos/opt/default.nix
new file mode 100644
index 0000000..c9b6892
--- /dev/null
+++ b/nixos/opt/default.nix
@@ -0,0 +1,9 @@
+{ nvidia, llm, ... }:
+{
+  imports =
+    (if nvidia then [
+      ./nvidia.nix
+    ] else []) ++ (if llm then [
+      ./llm.nix
+    ] else []);
+}
diff --git a/nixos/opt/llm.nix b/nixos/opt/llm.nix
new file mode 100644
index 0000000..a944d5c
--- /dev/null
+++ b/nixos/opt/llm.nix
@@ -0,0 +1,7 @@
+{ config, pkgs, ... }:
+{
+  services.ollama = {
+    enable = true;
+    acceleration = "cuda";
+  };
+}
diff --git a/nixos/opt/nvidia.nix b/nixos/opt/nvidia.nix
new file mode 100644
index 0000000..63db5b2
--- /dev/null
+++ b/nixos/opt/nvidia.nix
@@ -0,0 +1,13 @@
+{ config, pkgs, ... }:
+{
+  hardware.graphics.enable = true;
+  hardware.nvidia = {
+    modesetting.enable = true;
+    package = config.boot.kernelPackages.nvidiaPackages.stable;
+    nvidiaSettings = true;
+    open = true;
+  };
+  boot.kernelParams = [ "acpi_backlight=video" ];
+  boot.initrd.kernelModules = [ "nvidia" "hp_wmi" ];
+  services.xserver.videoDrivers = [ "nvidia" ];
+}