34 lines
797 B
Nix
34 lines
797 B
Nix
# Run with `nix-shell cuda-fhs.nix`
|
|
{ pkgs ? import <nixpkgs> {} }:
|
|
(pkgs.buildFHSUserEnv {
|
|
name = "cuda-env";
|
|
targetPkgs = pkgs: with pkgs; [
|
|
git
|
|
gitRepo
|
|
gnupg
|
|
autoconf
|
|
curl
|
|
procps
|
|
gnumake
|
|
util-linux
|
|
m4
|
|
gperf
|
|
unzip
|
|
cudaPackages_12.cudatoolkit
|
|
linuxPackages.nvidia_x11
|
|
libGLU libGL
|
|
xorg.libXi xorg.libXmu freeglut
|
|
xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
|
|
ncurses5
|
|
stdenv.cc
|
|
binutils
|
|
];
|
|
multiPkgs = pkgs: with pkgs; [ zlib ];
|
|
runScript = "bash";
|
|
profile = ''
|
|
export CUDA_PATH=${pkgs.cudatoolkit}
|
|
# export LD_LIBRARY_PATH=${pkgs.linuxPackages.nvidia_x11}/lib
|
|
export EXTRA_LDFLAGS="-L/lib -L${pkgs.linuxPackages.nvidia_x11}/lib"
|
|
export EXTRA_CCFLAGS="-I/usr/include"
|
|
'';
|
|
}).env |