From 4eda7cf9e877d08daf4f26e87097354ec2a1df00 Mon Sep 17 00:00:00 2001 From: Someone Serge Date: Wed, 6 Apr 2022 00:50:25 +0300 Subject: [PATCH] cudnn 8.3.2: unbreak for cuda 11.6 It has been added to the official support matrix: https://docs.nvidia.com/deeplearning/cudnn/support-matrix/#cudnn-cuda-hardware-versions --- pkgs/development/libraries/science/math/cudnn/default.nix | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pkgs/development/libraries/science/math/cudnn/default.nix b/pkgs/development/libraries/science/math/cudnn/default.nix index a3f785c5b38f..d07f7e6c0b2e 100644 --- a/pkgs/development/libraries/science/math/cudnn/default.nix +++ b/pkgs/development/libraries/science/math/cudnn/default.nix @@ -15,6 +15,7 @@ , cudatoolkit_11_3 , cudatoolkit_11_4 , cudatoolkit_11_5 +, cudatoolkit_11_6 , fetchurl , lib }: @@ -89,7 +90,7 @@ rec { cudatoolkit = cudatoolkit_10_2; # See https://docs.nvidia.com/deeplearning/cudnn/archives/cudnn-832/support-matrix/index.html#cudnn-cuda-hardware-versions. minCudaVersion = "10.2.00000"; - maxCudaVersion = "11.5.99999"; + maxCudaVersion = "11.6.00001"; mkSrc = cudatoolkit: let v = if lib.versions.majorMinor cudatoolkit.version == "10.2" then "10.2" else "11.5"; in fetchurl { @@ -108,6 +109,7 @@ rec { cudnn_8_3_cudatoolkit_11_3 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_3; }; cudnn_8_3_cudatoolkit_11_4 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_4; }; cudnn_8_3_cudatoolkit_11_5 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_5; }; + cudnn_8_3_cudatoolkit_11_6 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_6; }; cudnn_8_3_cudatoolkit_10 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_10; }; cudnn_8_3_cudatoolkit_11 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11; };