From ec9d3ac349344670fea518c805e98aca3ee24007 Mon Sep 17 00:00:00 2001
From: Nick <nickjhiebert@proton.me>
Date: Fri, 18 Jul 2025 13:35:00 -0500
Subject: [PATCH] feat: added nvidia to server

---
 systems/ceres/config/graphics.nix | 67 ++++++++++++++++++++++++-------
 1 file changed, 52 insertions(+), 15 deletions(-)

diff --git a/systems/ceres/config/graphics.nix b/systems/ceres/config/graphics.nix
index b67a5df..091c3a5 100755
--- a/systems/ceres/config/graphics.nix
+++ b/systems/ceres/config/graphics.nix
@@ -1,24 +1,61 @@
-{ pkgs, ... }:
 {
+  pkgs,
+  config,
+  lib,
+  ...
+}:
+{
+  environment = {
+    etc."libva.conf".text = ''
+      LIBVA_MESSAGING_LEVEL=1
+    '';
+    systemPackages = builtins.attrValues {
+      inherit (pkgs) ddcutil;
+      # inherit (pkgs.cudaPackages) cudatoolkit cudnn; # cutensor; # tensorrt;
+      # inherit (pkgs.nvtopPackages) full;
+    };
+    sessionVariables = {
+      MOZ_DISABLE_RDD_SANDBOX = 1;
+      LIBVA_DRIVER_NAME = "nvidia";
+    };
+    variables = {
+      GBM_BACKEND = "nvidia-drm";
+      __GLX_VENDOR_LIBRARY_NAME = "nvidia";
+      NIZOS_OZONE_WL = "1";
+      KWIN_DRM_ALLOW_NVIDIA_COLORSPACE = "1";
+      ENABLE_HDR_WSI = "1";
+    };
+  };
+
   hardware = {
     graphics = {
       enable = true;
-      extraPackages = builtins.attrValues {
-        inherit (pkgs)
-          amdvlk
-          ;
-        inherit (pkgs.rocmPackages.clr)
-          icd
-          ;
-      };
+      enable32Bit = true;
+      extraPackages = builtins.attrValues { inherit (pkgs) nvidia-vaapi-driver vaapiVdpau; };
       extraPackages32 = builtins.attrValues {
-        inherit (pkgs.driversi686Linux)
-          amdvlk
-          ;
+        inherit (pkgs.pkgsi686Linux) nvidia-vaapi-driver vaapiVdpau;
+      };
+    };
+    nvidia = {
+      modesetting.enable = true;
+      forceFullCompositionPipeline = false;
+      nvidiaSettings = true;
+      open = true;
+      gsp.enable = true;
+      package = config.boot.kernelPackages.nvidiaPackages.beta;
+      powerManagement = {
+        enable = true;
+        finegrained = false;
       };
     };
   };
-  boot.initrd.kernelModules = [
-    "amdgpu"
-  ];
+
+  services = {
+    xserver = {
+      enable = true;
+      videoDrivers = [ "nvidia" ];
+    };
+  };
+
+  hardware.nvidia-container-toolkit.enable = lib.mkIf config.virtualisation.docker.enable true;
 }