Simplify flake.nix and use config.cudaSupport

Author: lukasl-dev

flake.nix

@@ -47,14 +47,6 @@
     }@inputs:
     let
       system = "x86_64-linux";
-      pkgs-unstable = import nixpkgs-unstable {
-        inherit system;
-        config.allowUnfree = true;
-        overlays = [ nixgl.overlay ];
-      };
-      specialArgs = {
-        inherit inputs pkgs-unstable;
-      };
 
       meta = {
         dir = "/home/lukas/nixos";
@@ -69,52 +61,56 @@
         time = {
           zone = "Europe/Vienna";
         };
+        cuda = false;
       };
-    in
-    {
-      nixosConfigurations = {
-        vega = nixpkgs.lib.nixosSystem {
-          inherit system;
-          specialArgs = specialArgs // {
-            meta = meta // {
-              hypr = {
-                monitors = [
-                  "DP-2, [email protected], 0x0, 1"
-                  "HDMI-A-1, [email protected], 1920x0, 1"
-                ];
-              };
+
+      nixosSystem =
+        module: overrideMeta:
+        let
+          updatedMeta = meta // overrideMeta;
+          pkgs-unstable = import nixpkgs-unstable {
+            inherit system;
+
+            config = {
+              allowUnfree = true;
+              cudaSupport = updatedMeta.cuda;
             };
-          };
-          modules = [ ./hosts/desktops/vega ];
-        };
 
-        orion = nixpkgs.lib.nixosSystem {
+            overlays = [ nixgl.overlay ];
+          };
+          specialArgs = {
+            inherit inputs pkgs-unstable;
+          };
+        in
+        nixpkgs.lib.nixosSystem {
           inherit system;
           specialArgs = specialArgs // {
-            meta = meta // {
-              hypr = {
-                monitors = [ "eDP-1, [email protected], 0x0, 1" ];
-              };
-            };
+            meta = updatedMeta;
           };
-          modules = [ ./hosts/desktops/orion ];
+          modules = [ module ];
         };
-
-        sirius = nixpkgs.lib.nixosSystem {
-          inherit system;
-          specialArgs = specialArgs // {
-            inherit meta;
+    in
+    {
+      nixosConfigurations = {
+        vega = nixosSystem ./hosts/desktops/vega {
+          cuda = true;
+          hypr = {
+            monitors = [
+              "DP-2, [email protected], 0x0, 1"
+              "HDMI-A-1, [email protected], 1920x0, 1"
+            ];
           };
-          modules = [ ./hosts/servers/sirius ];
         };
 
-        pollux = nixpkgs.lib.nixosSystem {
-          inherit system;
-          specialArgs = specialArgs // {
-            inherit meta;
+        orion = nixosSystem ./hosts/desktops/orion {
+          hypr = {
+            monitors = [ "eDP-1, [email protected], 0x0, 1" ];
           };
-          modules = [ ./hosts/servers/pollux ];
         };
+
+        sirius = nixosSystem ./hosts/servers/sirius { };
+
+        pollux = nixosSystem ./hosts/servers/pollux { };
       };
     };
 }

hosts/desktops/orion/default.nix

@@ -5,7 +5,7 @@
     ../default.nix
     ./hardware-configuration.nix
 
-    ../../../modules/ollama.nix
+    ../../../modules/llms.nix
   ];
 
   networking.hostName = "orion";

hosts/desktops/vega/default.nix

@@ -5,7 +5,7 @@
     ../default.nix
     ./hardware-configuration.nix
 
-    ../../../modules/ollama.nix
+    ../../../modules/llms.nix
   ];
 
   networking.hostName = "vega";
@@ -56,52 +56,4 @@
   environment.systemPackages = [
     pkgs.ffmpeg
   ];
-
-  systemd =
-    let
-      script = channel: duration: ''
-        dir="/home/lukas/tutils/recordings/$(date +'%Y/%m/%d')/${channel}"
-        mkdir -p "$dir"
-
-        out_file="$dir/$(date +'%H-%M').ts"
-
-        printf "Recording ${duration} to %s\n" "$out_file"
-        printf "Started at %s\n" "$(date)"
-
-        ${pkgs.ffmpeg}/bin/ffmpeg -headers "Referer: https://tuwel.tuwien.ac.at\r\n" \
-               -t "${duration}" \
-               -i "https://live-cdn-2.video.tuwien.ac.at/lecturetube-live/${channel}/playlist.m3u8" \
-               -c copy "$out_file"
-
-        printf "Finished at %s\n" "$(date)"
-      '';
-
-      service = channel: duration: {
-        description = "lecture-${channel}-${duration}";
-        script = script channel duration;
-        serviceConfig = {
-          Type = "oneshot";
-          User = "lukas";
-        };
-      };
-
-      timer = unit: calendar: {
-        wantedBy = [ "timers.target" ];
-        timerConfig = {
-          OnCalendar = calendar;
-          Unit = unit;
-        };
-      };
-    in
-    {
-      services = {
-        "audimax-7200" = service "bau178a-gm-1-audi-max" "7200";
-      };
-
-      timers = {
-        "monday-08am" = timer "audimax-7200.service" "Mon 08:00";
-        "monday-12am" = timer "audimax-7200.service" "Mon 12:00";
-        "monday-14am" = timer "audimax-7200.service" "Mon 14:00";
-      };
-    };
 }

modules/desktop/hyprland.nix

@@ -34,7 +34,7 @@
 
       egl-wayland
 
-      ueberzugpp
+      (ueberzugpp.override { enableOpencv = false; }) # TODO: fix cuda issues
 
       hyprsunset
     ];

modules/ollama.nix renamed to modules/llms.nix

@@ -17,6 +17,10 @@ in
     };
   };
 
+  environment.systemPackages = [
+    pkgs-unstable.llama-cpp
+  ];
+
   # services.open-webui = {
   #   enable = true;
   #   package = pkgs-unstable.open-webui;

modules/nixos/nix.nix

@@ -35,6 +35,7 @@
 
   nixpkgs.config = {
     allowUnfree = true;
+    cudaSupport = meta.cuda;
   };
 
   programs.nix-ld = {