python.pkgs.tensorflow: fix build with cudatoolkit 10.2
This commit is contained in:
parent
e7e5fc9584
commit
9a982d3540
@ -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,
|
@ -124,6 +124,10 @@ let
|
|||||||
sha256 = "077cpj0kzyqxzdya1dwh8df17zfzhqn7c685hx6iskvw2979zg2n";
|
sha256 = "077cpj0kzyqxzdya1dwh8df17zfzhqn7c685hx6iskvw2979zg2n";
|
||||||
})
|
})
|
||||||
./lift-gast-restriction.patch
|
./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
|
# On update, it can be useful to steal the changes from gentoo
|
||||||
|
Loading…
Reference in New Issue
Block a user