diff options
| author | Andreas Rammhold <andreas@rammhold.de> | 2019-11-06 08:58:30 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-11-06 08:58:30 +0100 |
| commit | 1e72fa51580dcac77549ffd9e499db6aa74d3286 (patch) | |
| tree | 434512691598033e18939a97ee0e70cba0ac601e /pkgs/development/libraries/science/math/cudnn | |
| parent | ec6579437c7ee95591c22738635e075e316754b0 (diff) | |
| parent | 78aff37ff1132c80b4db5e9551ff4404889cf0f4 (diff) | |
Merge pull request #68954 from dguibert/dg/cudatoolkit_10_1
cudatoolkit: add 10.1.243
Diffstat (limited to 'pkgs/development/libraries/science/math/cudnn')
| -rw-r--r-- | pkgs/development/libraries/science/math/cudnn/default.nix | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/pkgs/development/libraries/science/math/cudnn/default.nix b/pkgs/development/libraries/science/math/cudnn/default.nix index a33e3ec20096..8cd74939959d 100644 --- a/pkgs/development/libraries/science/math/cudnn/default.nix +++ b/pkgs/development/libraries/science/math/cudnn/default.nix @@ -1,4 +1,4 @@ -{ callPackage, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8, cudatoolkit_9_0, cudatoolkit_9_1, cudatoolkit_9_2, cudatoolkit_10_0 }: +{ callPackage, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8, cudatoolkit_9_0, cudatoolkit_9_1, cudatoolkit_9_2, cudatoolkit_10_0, cudatoolkit_10_1 }: let generic = args: callPackage (import ./generic.nix (removeAttrs args ["cudatoolkit"])) { @@ -65,5 +65,12 @@ in rec { sha256 = "18ys0apiz9afid2s6lvy9qbyi8g66aimb2a7ikl1f3dm09mciprf"; }; - cudnn_cudatoolkit_10 = cudnn_cudatoolkit_10_0; + cudnn_cudatoolkit_10_1 = generic rec { + version = "7.6.3"; + cudatoolkit = cudatoolkit_10_1; + srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v7.6.3.30.tgz"; + sha256 = "0qc9f1xpyfibwqrpqxxq2v9h6w90j0dbx564akwy44c1dls5f99m"; + }; + + cudnn_cudatoolkit_10 = cudnn_cudatoolkit_10_1; } |