diff options
| author | David Guibert <david.guibert@gmail.com> | 2019-09-17 14:22:02 +0200 |
|---|---|---|
| committer | David Guibert <david.guibert@gmail.com> | 2019-09-17 15:19:41 +0200 |
| commit | 92690a89376f6f57136f93053308588eebd86dec (patch) | |
| tree | 78a3ea398aed6d1f1f22ed9e4be4ef4526c96769 | |
| parent | 4a3e7f83f7a210e3b57320e80d900f1f25b59ea8 (diff) | |
| download | nixpkgs-92690a89376f6f57136f93053308588eebd86dec.tar nixpkgs-92690a89376f6f57136f93053308588eebd86dec.tar.gz nixpkgs-92690a89376f6f57136f93053308588eebd86dec.tar.bz2 nixpkgs-92690a89376f6f57136f93053308588eebd86dec.tar.lz nixpkgs-92690a89376f6f57136f93053308588eebd86dec.tar.xz nixpkgs-92690a89376f6f57136f93053308588eebd86dec.tar.zst nixpkgs-92690a89376f6f57136f93053308588eebd86dec.zip | |
cuddn: add 10.1 (version 7.6.3)
| -rw-r--r-- | pkgs/development/libraries/science/math/cudnn/default.nix | 11 | ||||
| -rw-r--r-- | pkgs/top-level/all-packages.nix | 3 |
2 files changed, 11 insertions, 3 deletions
diff --git a/pkgs/development/libraries/science/math/cudnn/default.nix b/pkgs/development/libraries/science/math/cudnn/default.nix index a33e3ec2009..8cd74939959 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; } diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 6d68ea75083..a3f224a8f97 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -2488,7 +2488,8 @@ in cudnn_cudatoolkit_9_1 cudnn_cudatoolkit_9_2 cudnn_cudatoolkit_10 - cudnn_cudatoolkit_10_0; + cudnn_cudatoolkit_10_0 + cudnn_cudatoolkit_10_1; cudnn = cudnn_cudatoolkit_9; |