tensorflow: Backport upstream as bazel build failing for current

2018-08-10 10:38:47 -04:00
parent d0a6a1bf3f
commit b6f66a65f5
10 changed files with 451 additions and 0 deletions
--- a/temporary/tf-cudnn.nix
+++ b/temporary/tf-cudnn.nix
@@ -0,0 +1,17 @@
+{ 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";
+  };
+}