Unverified Commit 21cd7478 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

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

parents 70cef8a3 3c03aded

pkgs/development/python-modules/cupy/default.nix

+0 −3
Original line number Diff line number Diff line
@@ -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

  ];

pkgs/top-level/python-packages.nix

+1 −15
Original line number Diff line number Diff line
@@ -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 { };