18 lines
418 B
Nix
18 lines
418 B
Nix
{ callPackage, cudatoolkit90 }:
|
|
|
|
let
|
|
generic = args: callPackage (import ./tf-cudnn-generic.nix (removeAttrs args ["cudatoolkit"])) {
|
|
inherit (args) cudatoolkit;
|
|
};
|
|
|
|
in
|
|
|
|
{
|
|
cudnn_cudatoolkit90 = generic rec {
|
|
version = "7.0.5";
|
|
cudatoolkit = cudatoolkit90;
|
|
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v7.tgz";
|
|
sha256 = "03mbv4m5lhwnc181xz8li067pjzzhxqbxgnrfc68dffm8xj0fghs";
|
|
};
|
|
}
|