diff options
author | Timo Kaufmann <timokau@zoho.com> | 2019-11-19 22:45:06 +0100 |
---|---|---|
committer | John Ericson <John.Ericson@Obsidian.Systems> | 2020-03-30 13:58:28 +0000 |
commit | a730888ce0c608789fbc57da7449c6a618b3e906 (patch) | |
tree | 4b0be30dac533732877f0b90e9b1972640e0609b /pkgs/top-level/python-packages.nix | |
parent | d4de62dedd118b1e482307a1c8561985fee0fe01 (diff) | |
download | nixlib-a730888ce0c608789fbc57da7449c6a618b3e906.tar nixlib-a730888ce0c608789fbc57da7449c6a618b3e906.tar.gz nixlib-a730888ce0c608789fbc57da7449c6a618b3e906.tar.bz2 nixlib-a730888ce0c608789fbc57da7449c6a618b3e906.tar.lz nixlib-a730888ce0c608789fbc57da7449c6a618b3e906.tar.xz nixlib-a730888ce0c608789fbc57da7449c6a618b3e906.tar.zst nixlib-a730888ce0c608789fbc57da7449c6a618b3e906.zip |
python.pkgs.tensorflow{,-bin,-estimator,-tensorboard}-2: Init at 2.1.0
Major breaking change from 1.x, so treating keeping both versions for now. (adapted from 33f11be707e39acf96423f97f3baa80d8f11a0cb) (adapted from 9e8dea7986dbdde850a58c7704182776642d8919)
Diffstat (limited to 'pkgs/top-level/python-packages.nix')
-rw-r--r-- | pkgs/top-level/python-packages.nix | 42 |
1 files changed, 36 insertions, 6 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 648b815e9ea2..7d1892f2be11 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -6548,21 +6548,49 @@ in { zerobin = callPackage ../development/python-modules/zerobin { }; - tensorflow-estimator = callPackage ../development/python-modules/tensorflow-estimator { }; - tensorflow-estimator_1_15_1 = callPackage ../development/python-modules/tensorflow-estimator/1_15_1.nix { }; + tensorflow-estimator = self.tensorflow-estimator_1; + + tensorflow-estimator_1 = callPackage ../development/python-modules/tensorflow-estimator/1 { }; + + tensorflow-estimator_2 = callPackage ../development/python-modules/tensorflow-estimator/2 { }; tensorflow-probability = callPackage ../development/python-modules/tensorflow-probability { }; - tensorflow-tensorboard = callPackage ../development/python-modules/tensorflow-tensorboard { }; + tensorflow-tensorboard = self.tensorflow-tensorboard_1; + + tensorflow-tensorboard_1 = callPackage ../development/python-modules/tensorflow-tensorboard/1 { }; + + tensorflow-tensorboard_2 = callPackage ../development/python-modules/tensorflow-tensorboard/2 { }; + + tensorflow-bin = self.tensorflow-bin_1; + + tensorflow-bin_1 = callPackage ../development/python-modules/tensorflow/1/bin.nix { + cudaSupport = pkgs.config.cudaSupport or false; + inherit (pkgs.linuxPackages) nvidia_x11; + cudatoolkit = pkgs.cudatoolkit_10; + cudnn = pkgs.cudnn_cudatoolkit_10; + }; - tensorflow-bin = callPackage ../development/python-modules/tensorflow/bin.nix { + tensorflow-bin_2 = callPackage ../development/python-modules/tensorflow/2/bin.nix { cudaSupport = pkgs.config.cudaSupport or false; inherit (pkgs.linuxPackages) nvidia_x11; cudatoolkit = pkgs.cudatoolkit_10; cudnn = pkgs.cudnn_cudatoolkit_10; }; - tensorflow-build = callPackage ../development/python-modules/tensorflow { + tensorflow-build = self.tensorflow-build_1; + + tensorflow-build_1 = callPackage ../development/python-modules/tensorflow/1 { + cudaSupport = pkgs.config.cudaSupport or false; + inherit (pkgs.linuxPackages) nvidia_x11; + cudatoolkit = pkgs.cudatoolkit_10; + cudnn = pkgs.cudnn_cudatoolkit_10; + nccl = pkgs.nccl_cudatoolkit_10; + openssl = pkgs.openssl_1_1; + inherit (pkgs.darwin.apple_sdk.frameworks) Foundation Security; + }; + + tensorflow-build_2 = callPackage ../development/python-modules/tensorflow/2 { cudaSupport = pkgs.config.cudaSupport or false; inherit (pkgs.linuxPackages) nvidia_x11; cudatoolkit = pkgs.cudatoolkit_10; @@ -6572,7 +6600,9 @@ in { inherit (pkgs.darwin.apple_sdk.frameworks) Foundation Security; }; - tensorflow = self.tensorflow-build; + tensorflow = self.tensorflow_1; + tensorflow_1 = self.tensorflow-build_1; + tensorflow_2 = self.tensorflow-build_2; tensorflowWithoutCuda = self.tensorflow.override { cudaSupport = false; |