4 files changed, 73 insertions, 0 deletions
diff --git a/modules/desktop/software/video/default.nix b/modules/desktop/software/video/default.nix
new file mode 100644
index 0000000..280a7d9
--- /dev/null
+++ b/modules/desktop/software/video/default.nix
@@ -0,0 +1,7 @@
+{
+  imports = [
+    ./graphics.nix
+    ./libva.nix
+    ./nvidia.nix
+  ];
+}
diff --git a/modules/desktop/software/video/graphics.nix b/modules/desktop/software/video/graphics.nix
new file mode 100644
index 0000000..13da295
--- /dev/null
+++ b/modules/desktop/software/video/graphics.nix
@@ -0,0 +1,21 @@
+{ pkgs, ... }:
+{
+  hardware.graphics = {
+    enable = true;
+    enable32Bit = true;
+
+    extraPackages = with pkgs; [
+      nvidia-vaapi-driver
+      vaapiVdpau
+      libvdpau-va-gl
+    ];
+
+    extraPackages32 = with pkgs; [
+      nvidia-vaapi-driver
+      vaapiVdpau
+      libvdpau-va-gl
+    ];
+  };
+
+  environment.systemPackages = [ pkgs.mesa ];
+}
diff --git a/modules/desktop/software/video/libva.nix b/modules/desktop/software/video/libva.nix
new file mode 100644
index 0000000..d420495
--- /dev/null
+++ b/modules/desktop/software/video/libva.nix
@@ -0,0 +1,7 @@
+{ pkgs, ... }:
+{
+  environment.systemPackages = with pkgs; [
+    libva
+    libva-utils
+  ];
+}
diff --git a/modules/desktop/software/video/nvidia.nix b/modules/desktop/software/video/nvidia.nix
new file mode 100644
index 0000000..c133bc2
--- /dev/null
+++ b/modules/desktop/software/video/nvidia.nix
@@ -0,0 +1,38 @@
+{ pkgs, config, ... }:
+{
+  environment = {
+    systemPackages = with pkgs; [
+      nvidia-container-toolkit
+      nvidia-docker
+    ];
+
+    shellAliases.nvidia-settings = "nvidia-settings --config='$XDG_CONFIG_HOME'/nvidia/settings";
+  };
+
+  boot = {
+    blacklistedKernelModules = [ "nouveau" ];
+
+    kernelParams = [
+      "nvidia-drm.fbdev=1"
+      "nvidia-drm.modeset=1"
+      "nvidia.NVreg_PreserveVideoMemoryAllocations=1"
+    ];
+  };
+
+  hardware = {
+    nvidia-container-toolkit.enable = true;
+
+    nvidia = {
+      modesetting.enable = true;
+      open = false;
+      nvidiaSettings = true;
+      package = config.boot.kernelPackages.nvidiaPackages.production;
+      forceFullCompositionPipeline = true;
+
+      powerManagement = {
+        enable = true;
+        finegrained = false;
+      };
+    };
+  };
+}