···

       
8
       
 
       
`cudaPackages_11_6`. Within each set is a matching version of the above listed

     

       
9
       
 
       
packages. Additionally, other versions of the packages that are packaged and

     

       
10
       
 
       
compatible are available as well. For example, there can be a

     

       
11
       
-
       
`cudaPackages.cudnn_8_3_2` package.

     

       
12
       
 
       


     

       
13
       
 
       
To use one or more CUDA packages in an expression, give the expression a `cudaPackages` parameter, and in case CUDA is optional

     

       
14
       
 
       
```nix

     
···

       
28
       
 
       
```nix

     

       
29
       
 
       
mypkg = let

     

       
30
       
 
       
  cudaPackages = cudaPackages_11_5.overrideScope' (final: prev: {

     

       
31
       
-
       
    cudnn = prev.cudnn_8_3_2;

     

       
32
       
 
       
  }});

     

       
33
       
 
       
in callPackage { inherit cudaPackages; };

     

       
34
       
 
       
```

     

···

       
8
       
 
       
`cudaPackages_11_6`. Within each set is a matching version of the above listed

     

       
9
       
 
       
packages. Additionally, other versions of the packages that are packaged and

     

       
10
       
 
       
compatible are available as well. For example, there can be a

     

       
11
       
+
       
`cudaPackages.cudnn_8_3` package.

     

       
12
       
 
       


     

       
13
       
 
       
To use one or more CUDA packages in an expression, give the expression a `cudaPackages` parameter, and in case CUDA is optional

     

       
14
       
 
       
```nix

     
···

       
28
       
 
       
```nix

     

       
29
       
 
       
mypkg = let

     

       
30
       
 
       
  cudaPackages = cudaPackages_11_5.overrideScope' (final: prev: {

     

       
31
       
+
       
    cudnn = prev.cudnn_8_3;

     

       
32
       
 
       
  }});

     

       
33
       
 
       
in callPackage { inherit cudaPackages; };

     

       
34
       
 
       
```

     

···

       
27
       
 
       
  # Earlier versions of cudatoolkit use pre-8.x CUDNN, so we use the default.

     

       
28
       
 
       
  cudnn = if lib.versionOlder cudatoolkit.version "10.1"

     

       
29
       
 
       
    then cudaPackages.cudnn

     

       
30
       
-
       
    else cudaPackages.cudnn_7_6_5;

     

       
31
       
 
       
in

     

       
32
       
 
       


     

       
33
       
 
       
assert leveldbSupport -> (leveldb != null && snappy != null);

     

···

       
27
       
 
       
  # Earlier versions of cudatoolkit use pre-8.x CUDNN, so we use the default.

     

       
28
       
 
       
  cudnn = if lib.versionOlder cudatoolkit.version "10.1"

     

       
29
       
 
       
    then cudaPackages.cudnn

     

       
30
       
+
       
    else cudaPackages.cudnn_7_6;

     

       
31
       
 
       
in

     

       
32
       
 
       


     

       
33
       
 
       
assert leveldbSupport -> (leveldb != null && snappy != null);

     

···

       
11
       
 
       
final: prev: let

     

       
12
       
 
       
  inherit (final) callPackage;

     

       
13
       
 
       
  inherit (prev) cudaVersion;

     

       
14
       
-
       
  inherit (prev.lib) attrsets lists versions strings trivial;

     

       
15
       
-
       


     

       
16
       
-
       
  # Utilities

     

       
17
       
-
       
  # majorMinorPatch :: String -> String

     

       
18
       
-
       
  majorMinorPatch = (trivial.flip trivial.pipe) [

     

       
19
       
-
       
    (versions.splitVersion)

     

       
20
       
-
       
    (lists.take 3)

     

       
21
       
-
       
    (strings.concatStringsSep ".")

     

       
22
       
-
       
  ];

     

       
23
       
 
       


     

       
24
       
 
       
  # Compute versioned attribute name to be used in this package set

     

       
0
       
 
       

     

       
25
       
 
       
  # computeName :: String -> String

     

       
26
       
-
       
  computeName = version: "cudnn_${strings.replaceStrings ["."] ["_"] (majorMinorPatch version)}";

     

       
27
       
 
       


     

       
28
       
 
       
  # Check whether a CUDNN release supports our CUDA version

     

       
29
       
 
       
  # Thankfully we're able to do lexicographic comparison on the version strings

     

       
30
       
 
       
  # isSupported :: Release -> Bool

     

       
31
       
 
       
  isSupported = release:

     

       
32
       
-
       
    strings.versionAtLeast cudaVersion release.minCudaVersion

     

       
33
       
-
       
    && strings.versionAtLeast release.maxCudaVersion cudaVersion;

     

       
34
       
 
       


     

       
35
       
 
       
  # useCudatoolkitRunfile :: Bool

     

       
36
       
-
       
  useCudatoolkitRunfile = strings.versionOlder cudaVersion "11.3.999";

     

       
37
       
 
       


     

       
38
       
 
       
  # buildCuDnnPackage :: Release -> Derivation

     

       
39
       
 
       
  buildCuDnnPackage = callPackage ./generic.nix {inherit useCudatoolkitRunfile;};

     

···

       
11
       
 
       
final: prev: let

     

       
12
       
 
       
  inherit (final) callPackage;

     

       
13
       
 
       
  inherit (prev) cudaVersion;

     

       
14
       
+
       
  inherit (prev.lib) attrsets lists versions;

     

       
15
       
+
       
  inherit (prev.lib.strings) replaceStrings versionAtLeast versionOlder;

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
16
       
 
       


     

       
17
       
 
       
  # Compute versioned attribute name to be used in this package set

     

       
18
       
+
       
  # Patch version changes should not break the build, so we only use major and minor

     

       
19
       
 
       
  # computeName :: String -> String

     

       
20
       
+
       
  computeName = version: "cudnn_${replaceStrings ["."] ["_"] (versions.majorMinor version)}";

     

       
21
       
 
       


     

       
22
       
 
       
  # Check whether a CUDNN release supports our CUDA version

     

       
23
       
 
       
  # Thankfully we're able to do lexicographic comparison on the version strings

     

       
24
       
 
       
  # isSupported :: Release -> Bool

     

       
25
       
 
       
  isSupported = release:

     

       
26
       
+
       
    versionAtLeast cudaVersion release.minCudaVersion

     

       
27
       
+
       
    && versionAtLeast release.maxCudaVersion cudaVersion;

     

       
28
       
 
       


     

       
29
       
 
       
  # useCudatoolkitRunfile :: Bool

     

       
30
       
+
       
  useCudatoolkitRunfile = versionOlder cudaVersion "11.3.999";

     

       
31
       
 
       


     

       
32
       
 
       
  # buildCuDnnPackage :: Release -> Derivation

     

       
33
       
 
       
  buildCuDnnPackage = callPackage ./generic.nix {inherit useCudatoolkitRunfile;};

     

···

       
155
       
 
       
    hash = "sha256-l2xMunIzyXrnQAavq1Fyl2MAukD1slCiH4z3H1nJ920=";

     

       
156
       
 
       
  }

     

       
157
       
 
       
  {

     

       
158
       
-
       
    version = "8.8.0.121";

     

       
159
       
 
       
    minCudaVersion = "11.0";

     

       
160
       
 
       
    maxCudaVersion = "11.8";

     

       
161
       
-
       
    url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.8.0.121_cuda11-archive.tar.xz";

     

       
162
       
-
       
    hash = "sha256-YgRkdgdtG0VfsT+3izjTSWusr7/bsElPszkiQKBEZuo=";

     

       
163
       
 
       
  }

     

       
164
       
 
       
  {

     

       
165
       
-
       
    version = "8.8.0.121";

     

       
166
       
 
       
    minCudaVersion = "12.0";

     

       
167
       
 
       
    maxCudaVersion = "12.0";

     

       
168
       
-
       
    url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.8.0.121_cuda12-archive.tar.xz";

     

       
169
       
-
       
    hash = "sha256-oHkrZmyq9ZOp3UEwl5V4/Tp4Iw9EB2RcKVcA7456qvI=";

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
170
       
 
       
  }

     

       
171
       
 
       
]

     

···

       
155
       
 
       
    hash = "sha256-l2xMunIzyXrnQAavq1Fyl2MAukD1slCiH4z3H1nJ920=";

     

       
156
       
 
       
  }

     

       
157
       
 
       
  {

     

       
158
       
+
       
    version = "8.8.1.3";

     

       
159
       
 
       
    minCudaVersion = "11.0";

     

       
160
       
 
       
    maxCudaVersion = "11.8";

     

       
161
       
+
       
    url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.8.1.3_cuda11-archive.tar.xz";

     

       
162
       
+
       
    hash = "sha256-r3WEyuDMVSS1kT7wjCm6YVQRPGDrCjegWQqRtRWoqPk=";

     

       
163
       
 
       
  }

     

       
164
       
 
       
  {

     

       
165
       
+
       
    version = "8.8.1.3";

     

       
166
       
 
       
    minCudaVersion = "12.0";

     

       
167
       
 
       
    maxCudaVersion = "12.0";

     

       
168
       
+
       
    url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.8.1.3_cuda12-archive.tar.xz";

     

       
169
       
+
       
    hash = "sha256-edd6dpx+cXWrx7XC7VxJQUjAYYqGQThyLIh/lcYjd3w=";

     

       
170
       
+
       
  }

     

       
171
       
+
       
  {

     

       
172
       
+
       
    version = "8.9.1.23";

     

       
173
       
+
       
    minCudaVersion = "11.0";

     

       
174
       
+
       
    maxCudaVersion = "11.8";

     

       
175
       
+
       
    url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz";

     

       
176
       
+
       
    hash = "sha256-ptmIcmfihZDJ25XOZcvpamaN8DUjOLfTN+BTLe0zSFw=";

     

       
177
       
+
       
  }

     

       
178
       
+
       
  {

     

       
179
       
+
       
    version = "8.9.1.23";

     

       
180
       
+
       
    minCudaVersion = "12.0";

     

       
181
       
+
       
    maxCudaVersion = "12.1";

     

       
182
       
+
       
    url = "https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.1.23_cuda12-archive.tar.xz";

     

       
183
       
+
       
    hash = "sha256-NRY8XFQr4MURc4sn4lI1GTy+7cXg4AbkSxzerxki6D4=";

     

       
184
       
 
       
  }

     

       
185
       
 
       
]