diff options
Diffstat (limited to 'hosts/poweredge/nvidia.nix')
| -rw-r--r-- | hosts/poweredge/nvidia.nix | 13 |
1 files changed, 10 insertions, 3 deletions
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 |