diff options
author | Nikolay Amiantov <ab@fmap.me> | 2018-03-15 03:08:06 +0300 |
---|---|---|
committer | Nikolay Amiantov <ab@fmap.me> | 2018-04-05 15:00:18 +0300 |
commit | 650d8c93a16333e9390db0269327bd2d47f02fc7 (patch) | |
tree | 680ac7effbaf603bf110eaf8da822213160fad23 /pkgs/top-level | |
parent | 35730455413fd0cd71d879ab9fd82f78f8e3f497 (diff) | |
download | nixlib-650d8c93a16333e9390db0269327bd2d47f02fc7.tar nixlib-650d8c93a16333e9390db0269327bd2d47f02fc7.tar.gz nixlib-650d8c93a16333e9390db0269327bd2d47f02fc7.tar.bz2 nixlib-650d8c93a16333e9390db0269327bd2d47f02fc7.tar.lz nixlib-650d8c93a16333e9390db0269327bd2d47f02fc7.tar.xz nixlib-650d8c93a16333e9390db0269327bd2d47f02fc7.tar.zst nixlib-650d8c93a16333e9390db0269327bd2d47f02fc7.zip |
tensorflow: add a binary wheel release for Darwin
Diffstat (limited to 'pkgs/top-level')
-rw-r--r-- | pkgs/top-level/python-packages.nix | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7d45b392792e..27928b76d0d0 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -19779,12 +19779,15 @@ EOF tensorflow-tensorboard = callPackage ../development/python-modules/tensorflow-tensorboard { }; - tensorflow = callPackage ../development/python-modules/tensorflow rec { - cudaSupport = pkgs.config.cudaSupport or false; - inherit (pkgs.linuxPackages) nvidia_x11; - cudatoolkit = pkgs.cudatoolkit9; - cudnn = pkgs.cudnn_cudatoolkit9; - }; + tensorflow = + if stdenv.isDarwin + then callPackage ../development/python-modules/tensorflow/bin.nix { } + else callPackage ../development/python-modules/tensorflow rec { + cudaSupport = pkgs.config.cudaSupport or false; + inherit (pkgs.linuxPackages) nvidia_x11; + cudatoolkit = pkgs.cudatoolkit9; + cudnn = pkgs.cudnn_cudatoolkit9; + }; tensorflowWithoutCuda = self.tensorflow.override { cudaSupport = false; |