Add option to enable beta driver

nikitawootten · Dec 2, 2023 · 7fe6438 · 7fe6438
1 parent 6f015fc
commit 7fe6438
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/hostModules/personal/nvidia.nix b/hostModules/personal/nvidia.nix
@@ -10,6 +10,7 @@ in
       description = "If true, enable NVidia settings";
       default = true;
     };
+    betaDriver = lib.mkEnableOption "enable beta driver";
   };
 
   config = lib.mkIf cfg.enable {
@@ -27,9 +28,11 @@ in
       modesetting.enable = true;
       open = false;
       nvidiaSettings = !cfg.headless;
-      package = config.boot.kernelPackages.nvidiaPackages.stable;
+      package = if cfg.betaDriver
+        then config.boot.kernelPackages.nvidiaPackages.beta
+        else config.boot.kernelPackages.nvidiaPackages.stable;
     };
 
     virtualisation.docker.enableNvidia = true;
   };
-}
+}