39 lines
969 B
Nix
39 lines
969 B
Nix
|
{
|
||
|
|
cudaAtLeast,
|
||
|
|
gmp,
|
||
|
|
expat,
|
||
|
|
libxcrypt-legacy,
|
||
|
|
ncurses6,
|
||
|
|
python310,
|
||
|
|
python311,
|
||
|
|
python312,
|
||
|
|
stdenv,
|
||
|
|
lib,
|
||
|
|
}:
|
||
|
|
prevAttrs: {
|
||
|
|
buildInputs =
|
||
|
|
prevAttrs.buildInputs or [ ]
|
||
|
|
# x86_64 only needs gmp from 12.0 and on
|
||
|
|
++ lib.lists.optionals (cudaAtLeast "12.0") [ gmp ]
|
||
|
|
# Additional dependencies for CUDA 12.5 and later, which
|
||
|
|
# support multiple Python versions.
|
||
|
|
++ lib.lists.optionals (cudaAtLeast "12.5") [
|
||
|
|
libxcrypt-legacy
|
||
|
|
ncurses6
|
||
|
|
python310
|
||
|
|
python311
|
||
|
|
python312
|
||
|
|
]
|
||
|
|
# aarch64,sbsa needs expat
|
||
|
|
++ lib.lists.optionals (stdenv.hostPlatform.isAarch64) [ expat ];
|
||
|
|
|
||
|
|
installPhase =
|
||
|
|
prevAttrs.installPhase or ""
|
||
|
|
# Python 3.8 is not in nixpkgs anymore, delete Python 3.8 cuda-gdb support
|
||
|
|
# to avoid autopatchelf failing to find libpython3.8.so.
|
||
|
|
+ lib.optionalString (cudaAtLeast "12.5") ''
|
||
|
|
find $bin -name '*python3.8*' -delete
|
||
|
|
find $bin -name '*python3.9*' -delete
|
||
|
|
'';
|
||
|
|
}
|