2024-06-17 14:58:37 +02:00
|
|
|
{ config, lib, pkgs, ... }:
|
|
|
|
{
|
|
|
|
# Enable OpenGL
|
|
|
|
hardware.opengl = {
|
|
|
|
enable = true;
|
|
|
|
driSupport = true;
|
|
|
|
driSupport32Bit = true;
|
|
|
|
};
|
2024-06-20 15:30:43 +02:00
|
|
|
nixpkgs.config = {
|
|
|
|
allowUnfree = true;
|
|
|
|
cudaSupport = true;
|
|
|
|
};
|
|
|
|
|
2024-06-17 14:58:37 +02:00
|
|
|
# Load nvidia driver for Xorg and Wayland
|
|
|
|
services.xserver.videoDrivers = ["nvidia"];
|
|
|
|
#boot with nvidia kernel module
|
|
|
|
boot.initrd.kernelModules = [ "nvidia" ];
|
|
|
|
hardware.nvidia = {
|
|
|
|
# Modesetting is required.
|
|
|
|
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 = true;
|
|
|
|
|
|
|
|
# Use the NVidia open source kernel module (not to be confused with the
|
|
|
|
# independent third-party "nouveau" open source driver).
|
|
|
|
# Support is limited to the Turing and later architectures. Full list of
|
|
|
|
# supported GPUs is at:
|
|
|
|
# https://github.com/NVIDIA/open-gpu-kernel-modules#compatible-gpus
|
|
|
|
# Only available from driver 515.43.04+
|
|
|
|
# Currently alpha-quality/buggy, so false is currently the recommended setting.
|
|
|
|
open = false;
|
|
|
|
|
|
|
|
# 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.cudnn
|
2024-06-18 16:30:12 +02:00
|
|
|
cudaPackages.cudatoolkit
|
2024-06-20 15:30:43 +02:00
|
|
|
cudaPackages.cudnn
|
2024-06-17 14:58:37 +02:00
|
|
|
nvtopPackages.nvidia
|
2024-06-20 15:30:43 +02:00
|
|
|
libxcrypt-legacy
|
|
|
|
gcc
|
|
|
|
xgboostWithCuda
|
|
|
|
cudaPackages.setupCudaHook
|
|
|
|
cudaPackages.nccl
|
|
|
|
cudaPackages.markForCudatoolkitRootHook
|
|
|
|
cmake
|
|
|
|
llama-cpp
|
|
|
|
python3Packages.pip
|
|
|
|
cudaPackages.cuda_cudart
|
|
|
|
cudaPackages.cuda_cudart.static
|
|
|
|
pkgs.cudaPackages.libcublas
|
|
|
|
|
2024-06-17 14:58:37 +02:00
|
|
|
#cudaPackages.tensorrt_8_6_0 #needs to be added manually, to the store and is a pain because of the license agreement and garbage collection
|
|
|
|
|
|
|
|
];
|
|
|
|
|
|
|
|
|
|
|
|
}
|