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,7 +15,10 @@ 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 ];
hardware = {
nvidia-container-toolkit.enable = true;
nvidia = {
# Modesetting is required. # Modesetting is required.
modesetting.enable = true; modesetting.enable = true;
# Nvidia power management. Experimental, and can cause sleep/suspend to fail. # Nvidia power management. Experimental, and can cause sleep/suspend to fail.
@@ -37,4 +41,21 @@ in
package = config.boot.kernelPackages.nvidiaPackages.stable; 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