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

python3Packages.tensorflow-build: enable on darwin

Browse Source
This commit is contained in:
Dmitry Kalinkin 2019-11-26 22:15:40 -05:00
parent 201defc4bf
commit 9c42df8d3e
No known key found for this signature in database
GPG Key ID: 5157B3EC8B2CA333
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkgs/development/python-modules/tensorflow/default.nix
View File

@ -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;

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

@ -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;