python3Packages.cupy: drop dependency on cudnn (#507780)

This commit is contained in:
Gaétan Lepage
2026-04-08 07:54:34 +00:00
committed by GitHub
2 changed files with 1 additions and 18 deletions
@@ -22,8 +22,6 @@
}:
let
inherit (cudaPackages) cudnn;
shouldUsePkg = lib.mapNullable (pkg: if pkg.meta.available or true then pkg else null);
# some packages are not available on all platforms
@@ -106,7 +104,6 @@ buildPythonPackage.override { stdenv = cudaPackages.backendStdenv; } (finalAttrs
buildInputs = [
cudatoolkit-joined
cudnn
libcutensor
nccl
];
+1 -15
View File
@@ -3396,21 +3396,7 @@ self: super: with self; {
cuda-pathfinder = callPackage ../development/python-modules/cuda-pathfinder { };
cupy = callPackage ../development/python-modules/cupy {
cudaPackages =
# CuDNN 9 is not supported:
# https://github.com/cupy/cupy/issues/8215
# NOTE: cupy 14 will drop support for cuDNN entirely.
# https://github.com/cupy/cupy/pull/9326
let
version = if pkgs.cudaPackages.backendStdenv.hasJetsonCudaCapability then "8.9.5" else "8.9.7";
in
pkgs.cudaPackages.override (prevArgs: {
manifests = prevArgs.manifests // {
cudnn = pkgs._cuda.manifests.cudnn.${version};
};
});
};
cupy = callPackage ../development/python-modules/cupy { };
curated-tokenizers = callPackage ../development/python-modules/curated-tokenizers { };