4eda7cf · typeable/nixpkgs

git.iko.soy/typeable/nixpkgs

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

commit4eda7cf9e877d08daf4f26e87097354ec2a1df00 authorSomeone Serge <sergei.kozlukov@aalto.fi> dateWed 6 Apr 2022, 00:50 (5 years ago) parent7e780e7

1 file changed+3−1

unifiedsplit

pkgs/development/libraries/science/math/cudnn/default.nix+3−1

@@ -15,6 +15,7 @@

1515, cudatoolkit_11_3

1616, cudatoolkit_11_4

1717, cudatoolkit_11_5

18+, cudatoolkit_11_6

1819, fetchurl

1920, lib

2021}:

@@ -89,7 +90,7 @@ rec {

8990 cudatoolkit = cudatoolkit_10_2;

9091 # See https://docs.nvidia.com/deeplearning/cudnn/archives/cudnn-832/support-matrix/index.html#cudnn-cuda-hardware-versions.

9192 minCudaVersion = "10.2.00000";

92− maxCudaVersion = "11.5.99999";

93+ maxCudaVersion = "11.6.00001";

9394 mkSrc = cudatoolkit:

9495 let v = if lib.versions.majorMinor cudatoolkit.version == "10.2" then "10.2" else "11.5"; in

9596 fetchurl {

@@ -108,6 +109,7 @@ rec {

108109 cudnn_8_3_cudatoolkit_11_3 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_3; };

109110 cudnn_8_3_cudatoolkit_11_4 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_4; };

110111 cudnn_8_3_cudatoolkit_11_5 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_5; };

112+ cudnn_8_3_cudatoolkit_11_6 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11_6; };

111113

112114 cudnn_8_3_cudatoolkit_10 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_10; };

113115 cudnn_8_3_cudatoolkit_11 = cudnn_8_3_cudatoolkit_10_2.override { cudatoolkit = cudatoolkit_11; };