diff options
author | Pavol Rusnak <pavol@rusnak.io> | 2020-01-04 17:34:29 +0100 |
---|---|---|
committer | Pavol Rusnak <pavol@rusnak.io> | 2020-01-06 13:46:00 +0100 |
commit | 9a982d3540497203632516616109459911129675 (patch) | |
tree | 94698a7a3898b7309fcfb53c658ae6ba9358bc83 /pkgs/development/python-modules/tensorflow | |
parent | e7e5fc9584864958281ff37c4ef46cd8c12fff0c (diff) | |
download | nixpkgs-9a982d3540497203632516616109459911129675.tar nixpkgs-9a982d3540497203632516616109459911129675.tar.gz nixpkgs-9a982d3540497203632516616109459911129675.tar.bz2 nixpkgs-9a982d3540497203632516616109459911129675.tar.lz nixpkgs-9a982d3540497203632516616109459911129675.tar.xz nixpkgs-9a982d3540497203632516616109459911129675.tar.zst nixpkgs-9a982d3540497203632516616109459911129675.zip |
python.pkgs.tensorflow: fix build with cudatoolkit 10.2
Diffstat (limited to 'pkgs/development/python-modules/tensorflow')
-rw-r--r-- | pkgs/development/python-modules/tensorflow/cuda-10.2-no-bin2c-path.patch | 10 | ||||
-rw-r--r-- | pkgs/development/python-modules/tensorflow/default.nix | 4 |
2 files changed, 14 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tensorflow/cuda-10.2-no-bin2c-path.patch b/pkgs/development/python-modules/tensorflow/cuda-10.2-no-bin2c-path.patch new file mode 100644 index 00000000000..25f1ba7a551 --- /dev/null +++ b/pkgs/development/python-modules/tensorflow/cuda-10.2-no-bin2c-path.patch @@ -0,0 +1,10 @@ +--- a/third_party/nccl/build_defs.bzl.tpl ++++ b/third_party/nccl/build_defs.bzl.tpl +@@ -113,7 +113,6 @@ def _device_link_impl(ctx): + "--cmdline=--compile-only", + "--link", + "--compress-all", +- "--bin2c-path=%s" % bin2c.dirname, + "--create=%s" % tmp_fatbin.path, + "--embedded-fatbin=%s" % fatbin_h.path, + ] + images, diff --git a/pkgs/development/python-modules/tensorflow/default.nix b/pkgs/development/python-modules/tensorflow/default.nix index 34282a5ec93..a9f3ebf9e48 100644 --- a/pkgs/development/python-modules/tensorflow/default.nix +++ b/pkgs/development/python-modules/tensorflow/default.nix @@ -124,6 +124,10 @@ let sha256 = "077cpj0kzyqxzdya1dwh8df17zfzhqn7c685hx6iskvw2979zg2n"; }) ./lift-gast-restriction.patch + + # cuda 10.2 does not have "-bin2c-path" option anymore + # https://github.com/tensorflow/tensorflow/issues/34429 + ./cuda-10.2-no-bin2c-path.patch ]; # On update, it can be useful to steal the changes from gentoo |