在 TensorFlow libdevice.10.bc InternalEr 中使用 XLA

Question

当我运行这行代码时：

tf.config.optimizer.set_jit(True)

要启用XLA in TensorFlow 2.1，我收到以下错误：

InternalError:  libdevice not found at ./libdevice.10.bc
     [[{{node cluster_1_1/xla_compile}}]]

我在这里查看了这篇文章： https: //github.com/google/jax/issues/989

但我似乎无法弄清楚如何通过Anaconda和使用来做到这一点os.environ['XLA_FLAGS']，而不是必须symlink command事先运行。

例如，我尝试过：

os.environ['XLA_FLAGS'] = "--xla_gpu_cuda_data_dir=/anaconda_path/envs/env/lib/libdevice.10.bc"

但我仍然遇到同样的错误。任何提示将非常感谢！

我在用着Ubuntu 18.04, TensorFlow 2.1, and Python 3.7.7。