python3Packages.cupy: use the latest cudnn

2022-04-01 08:02:47 +00:00
parent e51d84a97f
commit 2a0acdbe9f
1 changed files with 1 additions and 1 deletions
@@ -1961,7 +1961,7 @@ in {

  cupy = callPackage ../development/python-modules/cupy {
    cudatoolkit = pkgs.cudatoolkit_11;
-    cudnn = pkgs.cudnn_8_1_cudatoolkit_11;
+    cudnn = pkgs.cudnn_8_3_cudatoolkit_11;
    nccl = pkgs.nccl_cudatoolkit_11;
    cutensor = pkgs.cutensor_cudatoolkit_11;
  };