jolheiser/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tensorflow: expose binary builds

Browse Source
This commit is contained in:
Nikolay Amiantov 2019-07-31 13:33:26 +03:00
parent 19cdfe87f9
commit cd0e461f93
2 changed files with 24 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pkgs/top-level/all-packages.nix
View File

@ -1093,15 +1093,17 @@ in
tensor = libsForQt5.callPackage ../applications/networking/instant-messengers/tensor { }; tensor = libsForQt5.callPackage ../applications/networking/instant-messengers/tensor { };
libtensorflow-bin = callPackage ../development/libraries/science/math/tensorflow/bin.nix {
cudaSupport = pkgs.config.cudaSupport or false;
inherit (linuxPackages) nvidia_x11;
cudatoolkit = cudatoolkit_10_0;
cudnn = cudnn_cudatoolkit_10_0;
};
libtensorflow = libtensorflow =
if python.pkgs.tensorflow ? libtensorflow if python.pkgs.tensorflow ? libtensorflow
then python.pkgs.tensorflow.libtensorflow then python.pkgs.tensorflow.libtensorflow
else callPackage ../development/libraries/science/math/tensorflow/bin.nix { else libtensorflow-bin;
cudaSupport = pkgs.config.cudaSupport or false;
inherit (linuxPackages) nvidia_x11;
cudatoolkit = cudatoolkit_10_0;
cudnn = cudnn_cudatoolkit_10_0;
};
behdad-fonts = callPackage ../data/fonts/behdad-fonts { }; behdad-fonts = callPackage ../data/fonts/behdad-fonts { };

31
pkgs/top-level/python-packages.nix
View File

@ -5700,21 +5700,22 @@ in {
tensorflow-tensorboard = callPackage ../development/python-modules/tensorflow-tensorboard { }; tensorflow-tensorboard = callPackage ../development/python-modules/tensorflow-tensorboard { };
tensorflow = tensorflow-bin = callPackage ../development/python-modules/tensorflow/bin.nix {
if stdenv.isDarwin cudaSupport = pkgs.config.cudaSupport or false;
then callPackage ../development/python-modules/tensorflow/bin.nix { inherit (pkgs.linuxPackages) nvidia_x11;
cudaSupport = pkgs.config.cudaSupport or false; cudatoolkit = pkgs.cudatoolkit_10;
inherit (pkgs.linuxPackages) nvidia_x11; cudnn = pkgs.cudnn_cudatoolkit_10;
cudatoolkit = pkgs.cudatoolkit_10; };
cudnn = pkgs.cudnn_cudatoolkit_10;
} tensorflow-build = callPackage ../development/python-modules/tensorflow {
else callPackage ../development/python-modules/tensorflow rec { cudaSupport = pkgs.config.cudaSupport or false;
cudaSupport = pkgs.config.cudaSupport or false; inherit (pkgs.linuxPackages) nvidia_x11;
inherit (pkgs.linuxPackages) nvidia_x11; cudatoolkit = pkgs.cudatoolkit_10;
cudatoolkit = pkgs.cudatoolkit_10; cudnn = pkgs.cudnn_cudatoolkit_10;
cudnn = pkgs.cudnn_cudatoolkit_10; nccl = pkgs.nccl_cudatoolkit_10;
nccl = pkgs.nccl_cudatoolkit_10; };
};
tensorflow = if stdenv.isDarwin then self.tensorflow-bin else self.tensorflow-build;
tensorflowWithoutCuda = self.tensorflow.override { tensorflowWithoutCuda = self.tensorflow.override {
cudaSupport = false; cudaSupport = false;