Syzygial/ccpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tensorflow: Backport upstream as bazel build failing for current

Browse Source
This commit is contained in:
Tyson Whitehead
2018-08-10 10:38:47 -04:00
parent d0a6a1bf3f
commit b6f66a65f5
10 changed files with 451 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
temporary/python-packages.nix
View File

@@ -1,3 +1,19 @@
pkgs: self: super: with self; {
# Backport upstream master tensorflow (bazel build is failing)
astunparse = callPackage ./tf-astunparse.nix { };
gast = callPackage ./tf-gast.nix { };
tensorflow-tensorboard = callPackage ./tf-tensorboard.nix {
protobuf = protobuf3_5;
};
tensorflow = callPackage ./tf.nix {
protobuf = protobuf3_5;
cudaSupport = pkgs.config.cudaSupport or false;
cudatoolkit = pkgs.cudatoolkit90;
cudnn = pkgs.cudnn_cudatoolkit90;
inherit (pkgs.linuxPackages) nvidia_x11;
};
}