From 2a0acdbe9f36e3dbf4e4cbcf867f2c1e990f781f Mon Sep 17 00:00:00 2001 From: Samuel Ainsworth Date: Fri, 1 Apr 2022 08:02:47 +0000 Subject: [PATCH] python3Packages.cupy: use the latest cudnn --- pkgs/top-level/python-packages.nix | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index ed68190015b4..6f2a347a6e01 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -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; };