diff --git a/pkgs/development/python-modules/tensorflow/default.nix b/pkgs/development/python-modules/tensorflow/default.nix index 79be6a60c42..cc64e49e1b7 100644 --- a/pkgs/development/python-modules/tensorflow/default.nix +++ b/pkgs/development/python-modules/tensorflow/default.nix @@ -25,6 +25,8 @@ , sse42Support ? builtins.elem (stdenv.hostPlatform.platform.gcc.arch or "default") ["westmere" "sandybridge" "ivybridge" "haswell" "broadwell" "skylake" "skylake-avx512"] , avx2Support ? builtins.elem (stdenv.hostPlatform.platform.gcc.arch or "default") [ "haswell" "broadwell" "skylake" "skylake-avx512"] , fmaSupport ? builtins.elem (stdenv.hostPlatform.platform.gcc.arch or "default") [ "haswell" "broadwell" "skylake" "skylake-avx512"] +# Darwin deps +, Foundation, Security }: assert cudaSupport -> nvidia_x11 != null @@ -157,6 +159,9 @@ let cudatoolkit cudnn nvidia_x11 + ] ++ lib.optionals stdenv.isDarwin [ + Foundation + Security ]; # arbitrarily set to the current latest bazel version, overly careful @@ -326,7 +331,7 @@ let homepage = http://tensorflow.org; license = licenses.asl20; maintainers = with maintainers; [ jyp abbradar ]; - platforms = platforms.linux; + platforms = with platforms; linux ++ darwin; # The py2 build fails due to some issue importing protobuf. Possibly related to the fix in # https://github.com/akesandgren/easybuild-easyblocks/commit/1f2e517ddfd1b00a342c6abb55aef3fd93671a2b broken = !(xlaSupport -> cudaSupport) || !isPy3k; diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 86535df44f5..84f30bc1107 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -6180,6 +6180,7 @@ in { cudnn = pkgs.cudnn_cudatoolkit_10; nccl = pkgs.nccl_cudatoolkit_10; openssl = pkgs.openssl_1_0_2; + inherit (pkgs.darwin.apple_sdk.frameworks) Foundation Security; }; tensorflow = if stdenv.isDarwin then self.tensorflow-bin else self.tensorflow-build;