52 lines
1.5 KiB
Nix
52 lines
1.5 KiB
Nix
{
|
|
config,
|
|
lib,
|
|
pkgs,
|
|
...
|
|
}:
|
|
{
|
|
# Enable OpenGL
|
|
hardware.graphics = {
|
|
enable = true;
|
|
enable32Bit = true;
|
|
};
|
|
|
|
# Load nvidia driver for Xorg and Wayland
|
|
services.xserver.videoDrivers = [ "nvidia" ];
|
|
|
|
boot = {
|
|
blacklistedKernelModules = [ "nouveau" ];
|
|
extraModulePackages = [ config.boot.kernelPackages.nvidia_x11 ];
|
|
initrd.kernelModules = [ "nvidia" ];
|
|
};
|
|
|
|
hardware.nvidia = {
|
|
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).
|
|
# Currently alpha-quality/buggy, so false is currently the recommended setting.
|
|
open = false; # need proprietary for cuda.
|
|
|
|
# Enable the Nvidia settings menu, 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;
|
|
};
|
|
|
|
# Enable the CUDA toolkit
|
|
#install packages
|
|
environment.systemPackages = with pkgs; [
|
|
cudaPackages.cudatoolkit
|
|
cudaPackages.cudnn
|
|
nvtopPackages.nvidia
|
|
cudaPackages.nccl
|
|
pkgs.cudaPackages.libcublas
|
|
];
|
|
|
|
}
|