cudatoolkit: add 9.0

This version is a dependency of tensorflow binaries
This commit is contained in:
Jean-Philippe Bernardy
2018-05-31 22:21:54 +02:00
parent 9e730ee254
commit 397c4b7748
3 changed files with 18 additions and 2 deletions
@@ -1,4 +1,4 @@
{ callPackage, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8, cudatoolkit_9 }:
{ callPackage, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8, cudatoolkit_9_0, cudatoolkit_9 }:
let
generic = args: callPackage (import ./generic.nix (removeAttrs args ["cudatoolkit"])) {
@@ -36,6 +36,13 @@ in
sha256 = "9e0b31735918fe33a79c4b3e612143d33f48f61c095a3b993023cdab46f6d66e";
};
cudnn_cudatoolkit_9_0 = generic rec {
version = "7.0.5";
cudatoolkit = cudatoolkit_9_0;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v7.tgz";
sha256 = "03mbv4m5lhwnc181xz8li067pjzzhxqbxgnrfc68dffm8xj0fghs";
};
cudnn_cudatoolkit_9 = generic rec {
version = "7.0.5";
cudatoolkit = cudatoolkit_9;