23 lines
615 B
Nix
23 lines
615 B
Nix
{config, ... }: {
|
|
# Enable video drivers
|
|
services.xserver.videoDrivers = ["nvidia"];
|
|
|
|
hardware.nvidia = {
|
|
modesetting.enable = true;
|
|
nvidiaSettings = false;
|
|
package = config.boot.kernelPackages.nvidiaPackages.beta;
|
|
open = true;
|
|
};
|
|
|
|
pkgs.stdenv.mkDerivation {
|
|
name = "cuda-env-shell";
|
|
buildInputs = with pkgs; [
|
|
cudatoolkit
|
|
];
|
|
shellHook = ''
|
|
export CUDA_PATH=${pkgs.cudatoolkit}
|
|
export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
|
|
export EXTRA_CCFLAGS="-I/usr/include"
|
|
'';
|
|
}
|