cuddn: add 10.1 (version 7.6.3)

This commit is contained in:
David Guibert 2019-09-17 14:22:02 +02:00
parent 4a3e7f83f7
commit 92690a8937
2 changed files with 11 additions and 3 deletions

View file

@ -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;
}

View file

@ -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;