From 3c03adedfb19bbe1e911dd20086c54f41f24ad69 Mon Sep 17 00:00:00 2001 From: stefan Date: Tue, 7 Apr 2026 15:57:14 -0700 Subject: [PATCH] python3Packages.cupy: drop dependency on cudnn From cupy 14.0.0 release notes: > All cuDNN-related functionality has been completely removed from CuPy. We recommend users who need to access cuDNN functionality from Python to consider using cuDNN Frontend instead. --- pkgs/development/python-modules/cupy/default.nix | 3 --- pkgs/top-level/python-packages.nix | 16 +--------------- 2 files changed, 1 insertion(+), 18 deletions(-) diff --git a/pkgs/development/python-modules/cupy/default.nix b/pkgs/development/python-modules/cupy/default.nix index 024c99843ba0..a1ebd8f5a3f9 100644 --- a/pkgs/development/python-modules/cupy/default.nix +++ b/pkgs/development/python-modules/cupy/default.nix @@ -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 ]; diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 71574797509c..9d4062cdd287 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -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 { };