Add Nvidia Docker toolkit support
This commit is contained in:
@@ -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;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user