Merge pull request #81843 from OmnipotentEntity/cudnn-7.6.5

cudnn_cudatoolkit_10_2: init at 7.6.5
This commit is contained in:
Mario Rodas 2020-03-10 07:19:50 -05:00 committed by GitHub
commit 99722af1d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 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, cudatoolkit_10_1 }:
{ callPackage, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8, cudatoolkit_9_0, cudatoolkit_9_1, cudatoolkit_9_2, cudatoolkit_10_0, cudatoolkit_10_1, cudatoolkit_10_2 }:
let
generic = args: callPackage (import ./generic.nix (removeAttrs args ["cudatoolkit"])) {
@ -72,5 +72,12 @@ in rec {
sha256 = "0qc9f1xpyfibwqrpqxxq2v9h6w90j0dbx564akwy44c1dls5f99m";
};
cudnn_cudatoolkit_10_2 = generic rec {
version = "7.6.5";
cudatoolkit = cudatoolkit_10_2;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v7.6.5.32.tgz";
sha256 = "084c13vzjdkb5s1996yilybg6dgav1lscjr1xdcgvlmfrbr6f0k0";
};
cudnn_cudatoolkit_10 = cudnn_cudatoolkit_10_1;
}

View File

@ -2691,7 +2691,8 @@ in
cudatoolkit_9_2
cudatoolkit_10
cudatoolkit_10_0
cudatoolkit_10_1;
cudatoolkit_10_1
cudatoolkit_10_2;
cudatoolkit = cudatoolkit_10;
@ -2707,7 +2708,8 @@ in
cudnn_cudatoolkit_9_2
cudnn_cudatoolkit_10
cudnn_cudatoolkit_10_0
cudnn_cudatoolkit_10_1;
cudnn_cudatoolkit_10_1
cudnn_cudatoolkit_10_2;
cudnn = cudnn_cudatoolkit_10;