python3Packages.cupy: drop dependency on cudnn (#507780)
This commit is contained in:
@@ -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
|
||||
];
|
||||
|
||||
@@ -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 { };
|
||||
|
||||
|
||||
Reference in New Issue
Block a user