diff options
author | Nikolay Amiantov <ab@fmap.me> | 2019-08-27 02:43:32 +0300 |
---|---|---|
committer | Nikolay Amiantov <ab@fmap.me> | 2019-08-27 02:43:48 +0300 |
commit | 71095460b94b6c382afe188d1caa0b7048030b8a (patch) | |
tree | 17ee7f43c3cb51d2561e9450932e7d789a4a4336 /pkgs/development/python-modules | |
parent | 8a866548cce841c742c9a01ff998740759940312 (diff) | |
download | nixpkgs-71095460b94b6c382afe188d1caa0b7048030b8a.tar nixpkgs-71095460b94b6c382afe188d1caa0b7048030b8a.tar.gz nixpkgs-71095460b94b6c382afe188d1caa0b7048030b8a.tar.bz2 nixpkgs-71095460b94b6c382afe188d1caa0b7048030b8a.tar.lz nixpkgs-71095460b94b6c382afe188d1caa0b7048030b8a.tar.xz nixpkgs-71095460b94b6c382afe188d1caa0b7048030b8a.tar.zst nixpkgs-71095460b94b6c382afe188d1caa0b7048030b8a.zip |
tensorflow: fix build
You know, Tensorflow.
Diffstat (limited to 'pkgs/development/python-modules')
-rw-r--r-- | pkgs/development/python-modules/tensorflow/default.nix | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/pkgs/development/python-modules/tensorflow/default.nix b/pkgs/development/python-modules/tensorflow/default.nix index 70488a0d9dc..899e8936843 100644 --- a/pkgs/development/python-modules/tensorflow/default.nix +++ b/pkgs/development/python-modules/tensorflow/default.nix @@ -118,7 +118,7 @@ let keras-applications # libs taken from system through the TF_SYS_LIBS mechanism - grpc + # grpc sqlite openssl jsoncpp @@ -169,7 +169,8 @@ let "flatbuffers" "gast_archive" "gif_archive" - "grpc" + # Lots of errors, requires an older version + # "grpc" "hwloc" "icu" "jpeg" @@ -245,8 +246,7 @@ let ''; # FIXME: Tensorflow uses dlopen() for CUDA libraries. - # No idea why gpr isn't linked properly; perhaps Tensorflow expects a static library? - NIX_LDFLAGS = [ "-lgpr" ] ++ lib.optionals cudaSupport [ "-lcudart" "-lcublas" "-lcufft" "-lcurand" "-lcusolver" "-lcusparse" "-lcudnn" ]; + NIX_LDFLAGS = lib.optionals cudaSupport [ "-lcudart" "-lcublas" "-lcufft" "-lcurand" "-lcusolver" "-lcusparse" "-lcudnn" ]; hardeningDisable = [ "format" ]; @@ -261,15 +261,14 @@ let bazelTarget = "//tensorflow/tools/pip_package:build_pip_package //tensorflow/tools/lib_package:libtensorflow"; fetchAttrs = { - preInstall = '' - rm -rf $bazelOut/external/{bazel_tools,\@bazel_tools.marker,local_*,\@local_*} - ''; + # So that checksums don't depend on these. + TF_SYSTEM_LIBS = null; # cudaSupport causes fetch of ncclArchive, resulting in different hashes sha256 = if cudaSupport then - "19ll3f1i5qzd7ngz3m2jbxzgcrdjx5sv6kv2j5mcb8g3xsws8j5x" + "196pm3ynfafqlcxah07hkvphf536hpix1ydgsynr1yg08aynlvvx" else - "0y9kw3k4yvrxwdy7zry7nip9mdiwyv35r6mx65g4w7qajiypfc7i"; + "138r85n27ijzwxfwb5pcfyb79v14368jpckw0vmciz6pwf11bd9g"; }; buildAttrs = { |