ModuleNotFoundError：没有名为“tensorflow.python.trackable”的模块

Question

声明是：

from tensorflow.python.trackable import base as trackable

但出现模块未找到错误。

##### File "C:\Users\mahit\anaconda3\lib\site-packages\keras\dtensor\lazy_variable.py", line 26, in <module>
   from tensorflow.python.trackable import base as trackable

ModuleNotFoundError: No module named 'tensorflow.python.trackable'

该代码位于 keras 库内。我不明白为什么会出现此模块未找到错误。因为这是包装中附带的代码。也不确定 trackable 的作用。请帮我解决这个问题。

Answer 1

我摆脱了这个错误：卸载tensorflow并安装tf-nightly。尽管如此，我还是收到了另一个错误： Can't find libdevice directory ${CUDA_DIR}/nvvm/libdevice

在使用 Linux 操作系统的 Python 代码中，可以使用以下命令解决该错误： os.environ['XLA_FLAGS'] = '--xla_gpu_cuda_data_dir=/usr/lib/cuda/'