lol

Merge pull request #146285 from SuperSandro2000/tensorflow-bin

python39Packages.tensorflow: cleanups

authored by

Dmitry Kalinkin and committed by
GitHub
2e7f48aa fc4ff32d

+3 -9
+3 -9
pkgs/development/python-modules/tensorflow/bin.nix
··· 20 20 , tensorflow-estimator_2 21 21 , tensorflow-tensorboard_2 22 22 , cudaSupport ? false 23 - , cudatoolkit ? null 24 - , cudnn ? null 23 + , cudatoolkit 24 + , cudnn 25 25 , zlib 26 26 , python 27 27 , keras-applications ··· 37 37 # - the source build doesn't work on Darwin. 38 38 # - the source build is currently brittle and not easy to maintain 39 39 40 - assert cudaSupport -> cudatoolkit != null 41 - && cudnn != null; 42 - 43 40 # unsupported combination 44 41 assert ! (stdenv.isDarwin && cudaSupport); 45 42 46 43 let 47 44 packages = import ./binary-hashes.nix; 48 - 49 - variant = if cudaSupport then "-gpu" else ""; 50 - pname = "tensorflow${variant}"; 51 45 metadataPatch = ./relax-dependencies-metadata.patch; 52 46 patch = ./relax-dependencies.patch; 53 47 in buildPythonPackage { 54 - inherit pname; 48 + pname = "tensorflow" + lib.optionalString cudaSupport "-gpu"; 55 49 inherit (packages) version; 56 50 format = "wheel"; 57 51