Commit 45892b6e authored by Daniël de Kok's avatar Daniël de Kok
Browse files

python3Packages.cupy: use CuDNN 8

CuDNN 9 is not supported:
https://github.com/cupy/cupy/issues/8215
parent 3772ad92
Loading
Loading
Loading
Loading
+4 −2
Original line number Diff line number Diff line
@@ -17,7 +17,7 @@
}:

let
  inherit (cudaPackages) cudnn cutensor nccl;
  inherit (cudaPackages) cudnn_8_9 cutensor nccl;
  outpaths = with cudaPackages; [
      cuda_cccl # <nv/target>
      cuda_cudart
@@ -83,7 +83,9 @@ buildPythonPackage rec {

  buildInputs = [
    cudatoolkit-joined
    cudnn
    # CuDNN 9 is not supported:
    # https://github.com/cupy/cupy/issues/8215
    cudnn_8_9
    cutensor
    nccl
  ];