Jimbo/NixOS-Config
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Allow Cuda to be used by Nvidia clients

Browse source
This commit is contained in:
Jimbo 2024-08-26 15:40:42 -04:00
parent 5e5432dfdd
commit 08948089a3
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
nixos/hardware/nvidia.nix
View file

@ -8,4 +8,15 @@
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"
'';
}