tensorflow / addons

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
Apache License 2.0
1.69k stars 610 forks source link

Fix local build with GPU support #2764

Closed shkarupa-alex closed 1 year ago

shkarupa-alex commented 1 year ago

Allow local build with GPU support against TF 2.10

Partly fixes #2712 To build locally it still required to manually replace crosstool_top value in .bazelrc with "@local_config_cuda//crosstool:toolchain"

Description

Brief Description of the PR:

Fixes # (issue)

Type of change

Checklist:

How Has This Been Tested?

Local build with GPU support finished successful with these changes