70 lines
2.0 KiB
Nix
Executable File
70 lines
2.0 KiB
Nix
Executable File
{
|
|
lib,
|
|
config,
|
|
namespace,
|
|
...
|
|
}:
|
|
with lib;
|
|
let
|
|
inherit (lib.${namespace}) mkBoolOpt;
|
|
cfg = config.${namespace}.hardware.nvidia;
|
|
in
|
|
{
|
|
options.${namespace}.hardware.nvidia = {
|
|
enable = mkEnableOption "nvidia hardware config";
|
|
|
|
enableOpen = mkBoolOpt false "Enable open source driver";
|
|
|
|
nvidiaSettings = mkBoolOpt false "Install nvidia settings";
|
|
|
|
enableBeta = mkBoolOpt false "Enable Beta drivers";
|
|
|
|
enableNvidiaDocker = mkBoolOpt false "Enable nvidia docker support";
|
|
};
|
|
|
|
config = mkIf cfg.enable {
|
|
hardware = {
|
|
# Nvidia
|
|
nvidia = {
|
|
package =
|
|
if cfg.enableBeta then
|
|
config.boot.kernelPackages.nvidiaPackages.beta
|
|
else
|
|
config.boot.kernelPackages.nvidiaPackages.latest;
|
|
|
|
# Modesetting is required.
|
|
modesetting.enable = true;
|
|
|
|
# Nvidia power management. Experimental, and can cause sleep/suspend to fail.
|
|
powerManagement.enable = true;
|
|
|
|
# Fine-grained power management. Turns off GPU when not in use.
|
|
# Experimental and only works on modern Nvidia GPUs (Turing or newer).
|
|
powerManagement.finegrained = false;
|
|
|
|
# Use the NVidia open source kernel module (not to be confused with the
|
|
# independent third-party "nouveau" open source driver).
|
|
# Support is limited to the Turing and later architectures. Full list of
|
|
# supported GPUs is at:
|
|
# https://github.com/NVIDIA/open-gpu-kernel-modules#compatible-gpus
|
|
# Only available from driver 515.43.04+
|
|
# Currently alpha-quality/buggy, so false is currently the recommended setting.
|
|
open = cfg.enableOpen;
|
|
|
|
# Enable the Nvidia settings menu,
|
|
# accessible via `nvidia-settings`.
|
|
inherit (cfg) nvidiaSettings;
|
|
};
|
|
};
|
|
|
|
# Services configs
|
|
services.xserver = {
|
|
# Load nvidia driver for Xorg and Wayland
|
|
videoDrivers = [ "nvidia" ];
|
|
};
|
|
|
|
# Virtualisation
|
|
hardware.nvidia-container-toolkit.enable = cfg.enableNvidiaDocker;
|
|
};
|
|
}
|