swendel/nixpkgs
1
0
Fork 0
mirror of https://github.com/SebastianWendel/nixpkgs.git synced 2024-09-21 04:49:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

cudnn_cudatoolkit: init 11_3 and 11_4

Browse source
This commit is contained in:
David Guibert 2021-08-17 08:38:15 +02:00
parent 5f7019ed8a
commit 659354f0b7
2 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pkgs/development/libraries/science/math/cudnn/default.nix
View file

@ -1,4 +1,6 @@
{ callPackage, cudatoolkit_10_0, cudatoolkit_10_1, cudatoolkit_10_2, cudatoolkit_11_0, cudatoolkit_11_1, cudatoolkit_11_2 }:
{ callPackage, cudatoolkit_10_0, cudatoolkit_10_1, cudatoolkit_10_2, cudatoolkit_11_0, cudatoolkit_11_1, cudatoolkit_11_2
, cudatoolkit_11_3, cudatoolkit_11_4
}:
let
generic = args: callPackage (import ./generic.nix (removeAttrs args ["cudatoolkit"])) {
@ -46,5 +48,13 @@ in rec {
cudatoolkit = cudatoolkit_11_2;
};
cudnn_cudatoolkit_11_3 = cudnn_cudatoolkit_11_0.override {
cudatoolkit = cudatoolkit_11_3;
};
cudnn_cudatoolkit_11_4 = cudnn_cudatoolkit_11_0.override {
cudatoolkit = cudatoolkit_11_4;
};
cudnn_cudatoolkit_11 = cudnn_cudatoolkit_11_2;
}

4
pkgs/top-level/all-packages.nix
View file

@ -4154,7 +4154,9 @@ with pkgs;
cudnn_cudatoolkit_11
cudnn_cudatoolkit_11_0
cudnn_cudatoolkit_11_1
cudnn_cudatoolkit_11_2;
cudnn_cudatoolkit_11_2
cudnn_cudatoolkit_11_3
cudnn_cudatoolkit_11_4;
cudnn = cudnn_cudatoolkit_10;