Add Nvidia Docker toolkit support

This commit is contained in:
2025-09-30 16:07:58 +02:00
parent a8ebaf2dd6
commit fe19fa616e
2 changed files with 50 additions and 25 deletions

View File

@@ -6,13 +6,17 @@
extraBackends = [ pkgs.sane-airscan ]; extraBackends = [ pkgs.sane-airscan ];
disabledDefaultBackends = [ "escl" ]; disabledDefaultBackends = [ "escl" ];
}; };
logitech.wireless.enable = false; logitech.wireless = {
logitech.wireless.enableGraphical = false; enable = false;
enableGraphical = false;
};
graphics.enable = true; graphics.enable = true;
enableRedistributableFirmware = true; enableRedistributableFirmware = true;
keyboard.qmk.enable = true; keyboard.qmk.enable = true;
bluetooth.enable = true; bluetooth = {
bluetooth.powerOnBoot = true; enable = true;
powerOnBoot = false;
};
}; };
local.hardware-clock.enable = false; local.hardware-clock.enable = false;
} }

View File

@@ -1,6 +1,7 @@
{ {
lib, lib,
config, config,
pkgs,
... ...
}: }:
with lib; with lib;
@@ -14,27 +15,47 @@ in
config = mkIf cfg.enable { config = mkIf cfg.enable {
services.xserver.videoDrivers = [ "nvidia" ]; services.xserver.videoDrivers = [ "nvidia" ];
hardware.nvidia = { environment.systemPackages = with pkgs; [ nvidia-docker ];
# Modesetting is required. hardware = {
modesetting.enable = true; nvidia-container-toolkit.enable = true;
# Nvidia power management. Experimental, and can cause sleep/suspend to fail. nvidia = {
powerManagement.enable = true; # Modesetting is required.
# Fine-grained power management. Turns off GPU when not in use. modesetting.enable = true;
# Experimental and only works on modern Nvidia GPUs (Turing or newer). # Nvidia power management. Experimental, and can cause sleep/suspend to fail.
powerManagement.finegrained = false; powerManagement.enable = true;
# Use the NVidia open source kernel module (not to be confused with the # Fine-grained power management. Turns off GPU when not in use.
# independent third-party "nouveau" open source driver). # Experimental and only works on modern Nvidia GPUs (Turing or newer).
# Support is limited to the Turing and later architectures. Full list of powerManagement.finegrained = false;
# supported GPUs is at: # Use the NVidia open source kernel module (not to be confused with the
# https://github.com/NVIDIA/open-gpu-kernel-modules#compatible-gpus # independent third-party "nouveau" open source driver).
# Only available from driver 515.43.04+ # Support is limited to the Turing and later architectures. Full list of
# Currently alpha-quality/buggy, so false is currently the recommended setting. # supported GPUs is at:
open = false; # https://github.com/NVIDIA/open-gpu-kernel-modules#compatible-gpus
# Enable the Nvidia settings menu, # Only available from driver 515.43.04+
# accessible via `nvidia-settings`. # Currently alpha-quality/buggy, so false is currently the recommended setting.
nvidiaSettings = true; open = false;
# Optionally, you may need to select the appropriate driver version for your specific GPU. # Enable the Nvidia settings menu,
package = config.boot.kernelPackages.nvidiaPackages.stable; # accessible via `nvidia-settings`.
nvidiaSettings = true;
# Optionally, you may need to select the appropriate driver version for your specific GPU.
package = config.boot.kernelPackages.nvidiaPackages.stable;
};
}; };
}; };
} }
# Making nvidia docker toolkit work:
#
# sudo nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml
#
# sudo nvidia-ctk cdi list
#
# sudo tee /etc/docker/daemon.json > /dev/null <<EOF
# {
# "features": {
# "cdi": true
# }
# }
# EOF
#
# docker run --device nvidia.com/gpu=all