ae91cbe6cc
GitOrigin-RevId: 536fe36e23ab0fc8b7f35c24603422eee9fc17a2
364 lines
15 KiB
Nix
364 lines
15 KiB
Nix
# This module provides the proprietary NVIDIA X11 / OpenGL drivers.
|
|
|
|
{ config, lib, pkgs, ... }:
|
|
|
|
with lib;
|
|
|
|
let
|
|
nvidia_x11 = let
|
|
drivers = config.services.xserver.videoDrivers;
|
|
isDeprecated = str: (hasPrefix "nvidia" str) && (str != "nvidia");
|
|
hasDeprecated = drivers: any isDeprecated drivers;
|
|
in if (hasDeprecated drivers) then
|
|
throw ''
|
|
Selecting an nvidia driver has been modified for NixOS 19.03. The version is now set using `hardware.nvidia.package`.
|
|
''
|
|
else if (elem "nvidia" drivers) then cfg.package else null;
|
|
|
|
enabled = nvidia_x11 != null;
|
|
cfg = config.hardware.nvidia;
|
|
|
|
pCfg = cfg.prime;
|
|
syncCfg = pCfg.sync;
|
|
offloadCfg = pCfg.offload;
|
|
primeEnabled = syncCfg.enable || offloadCfg.enable;
|
|
nvidiaPersistencedEnabled = cfg.nvidiaPersistenced;
|
|
in
|
|
|
|
{
|
|
imports =
|
|
[
|
|
(mkRenamedOptionModule [ "hardware" "nvidia" "optimus_prime" "enable" ] [ "hardware" "nvidia" "prime" "sync" "enable" ])
|
|
(mkRenamedOptionModule [ "hardware" "nvidia" "optimus_prime" "allowExternalGpu" ] [ "hardware" "nvidia" "prime" "sync" "allowExternalGpu" ])
|
|
(mkRenamedOptionModule [ "hardware" "nvidia" "optimus_prime" "nvidiaBusId" ] [ "hardware" "nvidia" "prime" "nvidiaBusId" ])
|
|
(mkRenamedOptionModule [ "hardware" "nvidia" "optimus_prime" "intelBusId" ] [ "hardware" "nvidia" "prime" "intelBusId" ])
|
|
];
|
|
|
|
options = {
|
|
hardware.nvidia.powerManagement.enable = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
description = ''
|
|
Experimental power management through systemd. For more information, see
|
|
the NVIDIA docs, on Chapter 21. Configuring Power Management Support.
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.powerManagement.finegrained = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
description = ''
|
|
Experimental power management of PRIME offload. For more information, see
|
|
the NVIDIA docs, chapter 22. PCI-Express runtime power management.
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.modesetting.enable = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
description = ''
|
|
Enable kernel modesetting when using the NVIDIA proprietary driver.
|
|
|
|
Enabling this fixes screen tearing when using Optimus via PRIME (see
|
|
<option>hardware.nvidia.prime.sync.enable</option>. This is not enabled
|
|
by default because it is not officially supported by NVIDIA and would not
|
|
work with SLI.
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.prime.nvidiaBusId = mkOption {
|
|
type = types.str;
|
|
default = "";
|
|
example = "PCI:1:0:0";
|
|
description = ''
|
|
Bus ID of the NVIDIA GPU. You can find it using lspci; for example if lspci
|
|
shows the NVIDIA GPU at "01:00.0", set this option to "PCI:1:0:0".
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.prime.intelBusId = mkOption {
|
|
type = types.str;
|
|
default = "";
|
|
example = "PCI:0:2:0";
|
|
description = ''
|
|
Bus ID of the Intel GPU. You can find it using lspci; for example if lspci
|
|
shows the Intel GPU at "00:02.0", set this option to "PCI:0:2:0".
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.prime.amdgpuBusId = mkOption {
|
|
type = types.str;
|
|
default = "";
|
|
example = "PCI:4:0:0";
|
|
description = ''
|
|
Bus ID of the AMD APU. You can find it using lspci; for example if lspci
|
|
shows the AMD APU at "04:00.0", set this option to "PCI:4:0:0".
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.prime.sync.enable = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
description = ''
|
|
Enable NVIDIA Optimus support using the NVIDIA proprietary driver via PRIME.
|
|
If enabled, the NVIDIA GPU will be always on and used for all rendering,
|
|
while enabling output to displays attached only to the integrated Intel GPU
|
|
without a multiplexer.
|
|
|
|
Note that this option only has any effect if the "nvidia" driver is specified
|
|
in <option>services.xserver.videoDrivers</option>, and it should preferably
|
|
be the only driver there.
|
|
|
|
If this is enabled, then the bus IDs of the NVIDIA and Intel GPUs have to be
|
|
specified (<option>hardware.nvidia.prime.nvidiaBusId</option> and
|
|
<option>hardware.nvidia.prime.intelBusId</option>).
|
|
|
|
If you enable this, you may want to also enable kernel modesetting for the
|
|
NVIDIA driver (<option>hardware.nvidia.modesetting.enable</option>) in order
|
|
to prevent tearing.
|
|
|
|
Note that this configuration will only be successful when a display manager
|
|
for which the <option>services.xserver.displayManager.setupCommands</option>
|
|
option is supported is used.
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.prime.sync.allowExternalGpu = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
description = ''
|
|
Configure X to allow external NVIDIA GPUs when using optimus.
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.prime.offload.enable = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
description = ''
|
|
Enable render offload support using the NVIDIA proprietary driver via PRIME.
|
|
|
|
If this is enabled, then the bus IDs of the NVIDIA and Intel GPUs have to be
|
|
specified (<option>hardware.nvidia.prime.nvidiaBusId</option> and
|
|
<option>hardware.nvidia.prime.intelBusId</option>).
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.nvidiaPersistenced = mkOption {
|
|
default = false;
|
|
type = types.bool;
|
|
description = ''
|
|
Update for NVIDA GPU headless mode, i.e. nvidia-persistenced. It ensures all
|
|
GPUs stay awake even during headless mode.
|
|
'';
|
|
};
|
|
|
|
hardware.nvidia.package = lib.mkOption {
|
|
type = lib.types.package;
|
|
default = config.boot.kernelPackages.nvidiaPackages.stable;
|
|
defaultText = "config.boot.kernelPackages.nvidiaPackages.stable";
|
|
description = ''
|
|
The NVIDIA X11 derivation to use.
|
|
'';
|
|
example = "config.boot.kernelPackages.nvidiaPackages.legacy340";
|
|
};
|
|
};
|
|
|
|
config = let
|
|
igpuDriver = if pCfg.intelBusId != "" then "modesetting" else "amdgpu";
|
|
igpuBusId = if pCfg.intelBusId != "" then pCfg.intelBusId else pCfg.amdgpuBusId;
|
|
in mkIf enabled {
|
|
assertions = [
|
|
{
|
|
assertion = with config.services.xserver.displayManager; gdm.nvidiaWayland -> cfg.modesetting.enable;
|
|
message = "You cannot use wayland with GDM without modesetting enabled for NVIDIA drivers, set `hardware.nvidia.modesetting.enable = true`";
|
|
}
|
|
|
|
{
|
|
assertion = primeEnabled -> pCfg.intelBusId == "" || pCfg.amdgpuBusId == "";
|
|
message = ''
|
|
You cannot configure both an Intel iGPU and an AMD APU. Pick the one corresponding to your processor.
|
|
'';
|
|
}
|
|
{
|
|
assertion = primeEnabled -> pCfg.nvidiaBusId != "" && (pCfg.intelBusId != "" || pCfg.amdgpuBusId != "");
|
|
message = ''
|
|
When NVIDIA PRIME is enabled, the GPU bus IDs must configured.
|
|
'';
|
|
}
|
|
{
|
|
assertion = offloadCfg.enable -> versionAtLeast nvidia_x11.version "435.21";
|
|
message = "NVIDIA PRIME render offload is currently only supported on versions >= 435.21.";
|
|
}
|
|
{
|
|
assertion = !(syncCfg.enable && offloadCfg.enable);
|
|
message = "Only one NVIDIA PRIME solution may be used at a time.";
|
|
}
|
|
{
|
|
assertion = !(syncCfg.enable && cfg.powerManagement.finegrained);
|
|
message = "Sync precludes powering down the NVIDIA GPU.";
|
|
}
|
|
{
|
|
assertion = cfg.powerManagement.enable -> offloadCfg.enable;
|
|
message = "Fine-grained power management requires offload to be enabled.";
|
|
}
|
|
];
|
|
|
|
# If Optimus/PRIME is enabled, we:
|
|
# - Specify the configured NVIDIA GPU bus ID in the Device section for the
|
|
# "nvidia" driver.
|
|
# - Add the AllowEmptyInitialConfiguration option to the Screen section for the
|
|
# "nvidia" driver, in order to allow the X server to start without any outputs.
|
|
# - Add a separate Device section for the Intel GPU, using the "modesetting"
|
|
# driver and with the configured BusID.
|
|
# - OR add a separate Device section for the AMD APU, using the "amdgpu"
|
|
# driver and with the configures BusID.
|
|
# - Reference that Device section from the ServerLayout section as an inactive
|
|
# device.
|
|
# - Configure the display manager to run specific `xrandr` commands which will
|
|
# configure/enable displays connected to the Intel iGPU / AMD APU.
|
|
|
|
services.xserver.useGlamor = mkDefault offloadCfg.enable;
|
|
|
|
services.xserver.drivers = let
|
|
in optional primeEnabled {
|
|
name = igpuDriver;
|
|
display = offloadCfg.enable;
|
|
modules = optional (igpuDriver == "amdgpu") [ pkgs.xorg.xf86videoamdgpu ];
|
|
deviceSection = ''
|
|
BusID "${igpuBusId}"
|
|
${optionalString syncCfg.enable ''Option "AccelMethod" "none"''}
|
|
'';
|
|
} ++ singleton {
|
|
name = "nvidia";
|
|
modules = [ nvidia_x11.bin ];
|
|
display = !offloadCfg.enable;
|
|
deviceSection = optionalString primeEnabled
|
|
''
|
|
BusID "${pCfg.nvidiaBusId}"
|
|
${optionalString syncCfg.allowExternalGpu "Option \"AllowExternalGpus\""}
|
|
${optionalString cfg.powerManagement.finegrained "Option \"NVreg_DynamicPowerManagement=0x02\""}
|
|
'';
|
|
screenSection =
|
|
''
|
|
Option "RandRRotation" "on"
|
|
${optionalString syncCfg.enable "Option \"AllowEmptyInitialConfiguration\""}
|
|
'';
|
|
};
|
|
|
|
services.xserver.serverLayoutSection = optionalString syncCfg.enable ''
|
|
Inactive "Device-${igpuDriver}[0]"
|
|
'' + optionalString offloadCfg.enable ''
|
|
Option "AllowNVIDIAGPUScreens"
|
|
'';
|
|
|
|
services.xserver.displayManager.setupCommands = optionalString syncCfg.enable ''
|
|
# Added by nvidia configuration module for Optimus/PRIME.
|
|
${pkgs.xorg.xrandr}/bin/xrandr --setprovideroutputsource ${igpuDriver} NVIDIA-0
|
|
${pkgs.xorg.xrandr}/bin/xrandr --auto
|
|
'';
|
|
|
|
environment.etc."nvidia/nvidia-application-profiles-rc" = mkIf nvidia_x11.useProfiles {
|
|
source = "${nvidia_x11.bin}/share/nvidia/nvidia-application-profiles-rc";
|
|
};
|
|
|
|
hardware.opengl.package = mkIf (!offloadCfg.enable) nvidia_x11.out;
|
|
hardware.opengl.package32 = mkIf (!offloadCfg.enable) nvidia_x11.lib32;
|
|
hardware.opengl.extraPackages = optional offloadCfg.enable nvidia_x11.out;
|
|
hardware.opengl.extraPackages32 = optional offloadCfg.enable nvidia_x11.lib32;
|
|
|
|
environment.systemPackages = [ nvidia_x11.bin nvidia_x11.settings ]
|
|
++ optionals nvidiaPersistencedEnabled [ nvidia_x11.persistenced ];
|
|
|
|
systemd.packages = optional cfg.powerManagement.enable nvidia_x11.out;
|
|
|
|
systemd.services = let
|
|
baseNvidiaService = state: {
|
|
description = "NVIDIA system ${state} actions";
|
|
|
|
path = with pkgs; [ kbd ];
|
|
serviceConfig = {
|
|
Type = "oneshot";
|
|
ExecStart = "${nvidia_x11.out}/bin/nvidia-sleep.sh '${state}'";
|
|
};
|
|
};
|
|
|
|
nvidiaService = sleepState: (baseNvidiaService sleepState) // {
|
|
before = [ "systemd-${sleepState}.service" ];
|
|
requiredBy = [ "systemd-${sleepState}.service" ];
|
|
};
|
|
|
|
services = (builtins.listToAttrs (map (t: nameValuePair "nvidia-${t}" (nvidiaService t)) ["hibernate" "suspend"]))
|
|
// {
|
|
nvidia-resume = (baseNvidiaService "resume") // {
|
|
after = [ "systemd-suspend.service" "systemd-hibernate.service" ];
|
|
requiredBy = [ "systemd-suspend.service" "systemd-hibernate.service" ];
|
|
};
|
|
};
|
|
in optionalAttrs cfg.powerManagement.enable services
|
|
// optionalAttrs nvidiaPersistencedEnabled {
|
|
"nvidia-persistenced" = mkIf nvidiaPersistencedEnabled {
|
|
description = "NVIDIA Persistence Daemon";
|
|
wantedBy = [ "multi-user.target" ];
|
|
serviceConfig = {
|
|
Type = "forking";
|
|
Restart = "always";
|
|
PIDFile = "/var/run/nvidia-persistenced/nvidia-persistenced.pid";
|
|
ExecStart = "${nvidia_x11.persistenced}/bin/nvidia-persistenced --verbose";
|
|
ExecStopPost = "${pkgs.coreutils}/bin/rm -rf /var/run/nvidia-persistenced";
|
|
};
|
|
};
|
|
};
|
|
|
|
systemd.tmpfiles.rules = optional config.virtualisation.docker.enableNvidia
|
|
"L+ /run/nvidia-docker/bin - - - - ${nvidia_x11.bin}/origBin"
|
|
++ optional (nvidia_x11.persistenced != null && config.virtualisation.docker.enableNvidia)
|
|
"L+ /run/nvidia-docker/extras/bin/nvidia-persistenced - - - - ${nvidia_x11.persistenced}/origBin/nvidia-persistenced";
|
|
|
|
boot.extraModulePackages = [ nvidia_x11.bin ];
|
|
|
|
# nvidia-uvm is required by CUDA applications.
|
|
boot.kernelModules = [ "nvidia-uvm" ] ++
|
|
optionals config.services.xserver.enable [ "nvidia" "nvidia_modeset" "nvidia_drm" ];
|
|
|
|
# If requested enable modesetting via kernel parameter.
|
|
boot.kernelParams = optional (offloadCfg.enable || cfg.modesetting.enable) "nvidia-drm.modeset=1"
|
|
++ optional cfg.powerManagement.enable "nvidia.NVreg_PreserveVideoMemoryAllocations=1";
|
|
|
|
services.udev.extraRules =
|
|
''
|
|
# Create /dev/nvidia-uvm when the nvidia-uvm module is loaded.
|
|
KERNEL=="nvidia", RUN+="${pkgs.runtimeShell} -c 'mknod -m 666 /dev/nvidiactl c $$(grep nvidia-frontend /proc/devices | cut -d \ -f 1) 255'"
|
|
KERNEL=="nvidia_modeset", RUN+="${pkgs.runtimeShell} -c 'mknod -m 666 /dev/nvidia-modeset c $$(grep nvidia-frontend /proc/devices | cut -d \ -f 1) 254'"
|
|
KERNEL=="card*", SUBSYSTEM=="drm", DRIVERS=="nvidia", RUN+="${pkgs.runtimeShell} -c 'mknod -m 666 /dev/nvidia%n c $$(grep nvidia-frontend /proc/devices | cut -d \ -f 1) %n'"
|
|
KERNEL=="nvidia_uvm", RUN+="${pkgs.runtimeShell} -c 'mknod -m 666 /dev/nvidia-uvm c $$(grep nvidia-uvm /proc/devices | cut -d \ -f 1) 0'"
|
|
KERNEL=="nvidia_uvm", RUN+="${pkgs.runtimeShell} -c 'mknod -m 666 /dev/nvidia-uvm-tools c $$(grep nvidia-uvm /proc/devices | cut -d \ -f 1) 0'"
|
|
'' + optionalString cfg.powerManagement.finegrained ''
|
|
# Remove NVIDIA USB xHCI Host Controller devices, if present
|
|
ACTION=="add", SUBSYSTEM=="pci", ATTR{vendor}=="0x10de", ATTR{class}=="0x0c0330", ATTR{remove}="1"
|
|
|
|
# Remove NVIDIA USB Type-C UCSI devices, if present
|
|
ACTION=="add", SUBSYSTEM=="pci", ATTR{vendor}=="0x10de", ATTR{class}=="0x0c8000", ATTR{remove}="1"
|
|
|
|
# Remove NVIDIA Audio devices, if present
|
|
ACTION=="add", SUBSYSTEM=="pci", ATTR{vendor}=="0x10de", ATTR{class}=="0x040300", ATTR{remove}="1"
|
|
|
|
# Enable runtime PM for NVIDIA VGA/3D controller devices on driver bind
|
|
ACTION=="bind", SUBSYSTEM=="pci", ATTR{vendor}=="0x10de", ATTR{class}=="0x030000", TEST=="power/control", ATTR{power/control}="auto"
|
|
ACTION=="bind", SUBSYSTEM=="pci", ATTR{vendor}=="0x10de", ATTR{class}=="0x030200", TEST=="power/control", ATTR{power/control}="auto"
|
|
|
|
# Disable runtime PM for NVIDIA VGA/3D controller devices on driver unbind
|
|
ACTION=="unbind", SUBSYSTEM=="pci", ATTR{vendor}=="0x10de", ATTR{class}=="0x030000", TEST=="power/control", ATTR{power/control}="on"
|
|
ACTION=="unbind", SUBSYSTEM=="pci", ATTR{vendor}=="0x10de", ATTR{class}=="0x030200", TEST=="power/control", ATTR{power/control}="on"
|
|
'';
|
|
|
|
boot.extraModprobeConfig = mkIf cfg.powerManagement.finegrained ''
|
|
options nvidia "NVreg_DynamicPowerManagement=0x02"
|
|
'';
|
|
|
|
boot.blacklistedKernelModules = [ "nouveau" "nvidiafb" ];
|
|
|
|
services.acpid.enable = true;
|
|
|
|
};
|
|
|
|
}
|