52 lines
1.1 KiB
Nix
52 lines
1.1 KiB
Nix
{
|
|
config,
|
|
pkgs,
|
|
...
|
|
}: {
|
|
services.xserver.videoDrivers = ["nvidia"];
|
|
|
|
hardware.nvidia = {
|
|
open = true;
|
|
modesetting.enable = true;
|
|
powerManagement.enable = false;
|
|
nvidiaSettings = true;
|
|
nvidiaPersistenced = true;
|
|
};
|
|
|
|
hardware.nvidia-container-toolkit.enable = true;
|
|
|
|
hardware.graphics = {
|
|
enable = true;
|
|
enable32Bit = true;
|
|
extraPackages = with pkgs; [nvidia-vaapi-driver];
|
|
};
|
|
|
|
programs.nix-ld = {
|
|
enable = true;
|
|
libraries = with pkgs; [
|
|
cudaPackages.cudatoolkit
|
|
cudaPackages.cudnn
|
|
cudaPackages.nccl
|
|
config.hardware.nvidia.package
|
|
];
|
|
};
|
|
|
|
environment.variables = {
|
|
__GLX_VENDOR_LIBRARY_NAME = "nvidia";
|
|
GBM_BACKEND = "nvidia-drm";
|
|
LIBVA_DRIVER_NAME = "nvidia";
|
|
CUDA_HOME = "${pkgs.cudaPackages.cudatoolkit}";
|
|
CPATH = "${pkgs.cudaPackages.cudatoolkit}/include";
|
|
};
|
|
|
|
environment.systemPackages = with pkgs; [
|
|
mesa-demos
|
|
vulkan-tools
|
|
vulkan-headers
|
|
vulkan-validation-layers
|
|
libva-utils
|
|
vdpauinfo
|
|
egl-wayland
|
|
cudaPackages.cudatoolkit
|
|
];
|
|
}
|