1 files changed, 13 insertions, 11 deletions
diff --git a/modules/hardware/nvidia.nix b/modules/hardware/nvidia.nix
index d6c16e2..31e2100 100644
--- a/modules/hardware/nvidia.nix
+++ b/modules/hardware/nvidia.nix
@@ -1,17 +1,19 @@
 { config, ... }:
 {
-  hardware.nvidia = {
-    modesetting.enable = true;
-    open = false;
-    nvidiaSettings = true;
-    package = config.boot.kernelPackages.nvidiaPackages.beta;
-    forceFullCompositionPipeline = true;
+  hardware = {
+    nvidia-container-toolkit.enable = true;
 
-    powerManagement = {
-      enable = true;
-      finegrained = false;
-    };
+    nvidia = {
+      modesetting.enable = true;
+      open = false;
+      nvidiaSettings = true;
+      package = config.boot.kernelPackages.nvidiaPackages.production;
+      forceFullCompositionPipeline = true;
 
-    prime.nvidiaBusId = "PCI:43:0:0";
+      powerManagement = {
+        enable = true;
+        finegrained = false;
+      };
+    };
   };
 }