mirror of
https://gitlab.com/ahoneybun/nix-configs.git
synced 2025-05-12 11:14:02 -06:00
19 lines
674 B
Nix
19 lines
674 B
Nix
{ pkgs ? import <nixpkgs> {} }:
|
|
pkgs.mkShell {
|
|
name = "cuda-env-shell";
|
|
buildInputs = with pkgs; [
|
|
git gitRepo gnupg autoconf curl
|
|
procps gnumake util-linux m4 gperf unzip
|
|
cudatoolkit linuxPackages.nvidia_x11
|
|
libGLU libGL
|
|
xorg.libXi xorg.libXmu freeglut
|
|
xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
|
|
ncurses5 stdenv.cc binutils
|
|
];
|
|
shellHook = ''
|
|
export CUDA_PATH=${pkgs.cudatoolkit}
|
|
# export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11}/lib:${pkgs.ncurses5}/lib
|
|
export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
|
|
export EXTRA_CCFLAGS="-I/usr/include"
|
|
'';
|
|
}
|