From 2beb167341f90c09425e25eb044d8103db509926 Mon Sep 17 00:00:00 2001
From: Tim Keller <tjk@tjkeller.xyz>
Date: Tue, 4 Aug 2026 12:29:17 -0500
Subject: use allowUnfreeCudaPredicate instead of blanket allowUnfree for
 poweredge

---
 hosts/poweredge/nvidia.nix | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'hosts/poweredge/nvidia.nix')

diff --git a/hosts/poweredge/nvidia.nix b/hosts/poweredge/nvidia.nix
index 257a46c..5eecc71 100644
--- a/hosts/poweredge/nvidia.nix
+++ b/hosts/poweredge/nvidia.nix
@@ -1,7 +1,14 @@
-{ config, pkgs, ... }: {
+{ lib, config, pkgs, ... }: {
 	# Configure nixpkgs for nvidia/cuda
-	nixpkgs.config.allowUnfree = true;
-	nixpkgs.config.cudaSupport = true;
+	nixpkgs.config = {
+		cudaSupport = true;
+		allowUnfreePredicate = pkg:
+			pkgs._cuda.lib.allowUnfreeCudaPredicate pkg
+			|| builtins.elem (lib.getName pkg) [
+				"nvidia-x11"
+				"nvidia-kernel-modules"
+			];
+	};
 
 	# NVIDIA
 	services.xserver.videoDrivers = [ "nvidia" ];  # xserver.videoDrivers does not imply X
-- 
cgit v1.2.3