20 lines
521 B
Nix
20 lines
521 B
Nix
{pkgs}:
|
|
{
|
|
|
|
shellHook = ''
|
|
export NVCC=${pkgs.cudatoolkit}/bin/nvcc
|
|
|
|
export CUDA_ROOT_PATH=${pkgs.cudatoolkit}
|
|
export CUDA_CXXFLAGS="-I${pkgs.openmpi.out}/include -I$CUDA_ROOT_PATH/include"
|
|
|
|
# cudart
|
|
export CUDA_LIB_PATH=${pkgs.cudatoolkit.lib}
|
|
export CUDA_OUT_PATH=${pkgs.cudatoolkit.out}
|
|
|
|
# in here is libcuda
|
|
export CUDA_X11LIB="${pkgs.linuxPackages.nvidia_x11}/lib"
|
|
export CUDA_LDFLAGS="-L$CUDA_X11LIB -L$CUDA_ROOT_PATH/lib -L$CUDA_LIB_PATH/lib -lcuda -lcudart -lcublas"
|
|
'';
|
|
|
|
}
|