Ensure kernel headers are properly installed for GPU drivers

Author: TriangularDev

Trixie/config/includes.chroot_after_packages/etc/trios/Install-GPU-Drivers.sh

@@ -100,6 +100,14 @@ if $HAS_INTEL; then install_intel; fi
 if $HAS_AMD; then install_amd; fi
 if $HAS_NVIDIA; then install_nvidia; fi
 
+# ─── Kernel Header + DKMS Rebuild (All GPUs) ─────────────────
+echo "[*] Ensuring kernel headers and DKMS modules are up to date..."
+apt-get install -y --install-recommends linux-headers-$(uname -r) dkms
+
+# Rebuild all DKMS modules (important for NVIDIA / hybrid systems)
+echo "[*] Rebuilding DKMS modules..."
+dkms autoinstall || echo "[!] DKMS build reported issues (check logs under /var/lib/dkms)."
+
 # ─── Hybrid Handling ─────────────────────────────────────────
 if ($HAS_NVIDIA && $HAS_INTEL) || ($HAS_NVIDIA && $HAS_AMD) || ($HAS_AMD && $HAS_INTEL); then
     configure_hybrid