ImportError：无法从“tensorflow.compat.v2.experimental”导入 Keras 导入名称“dtensor”

Question

导入 Keras 时出现以下错误。

ImportError: cannot import name 'dtensor' from 'tensorflow.compat.v2.experimental' (C:\Users\User\AppData\Roaming\Python\Python38\site-packages\tensorflow\_api\v2\compat\v2\experimental\__init__.py)

TensorFlow v. 2.6，Keras v. 2.6

有人有解决此错误的想法吗？ 我将不胜感激任何帮助！

Answer 1

DTensor 是 TensorFlow 2.9.0 版本的一部分。 要导入dtensor ，您可以将tensorflow 2.6升级到tensorflow 2.9 ，如下所示：

pip install --upgrade tensorflow==2.9

现在，您可以从tensorflow.experimental或tensorflow.keras导入dtensor ，如下所示：

#Using tensorflow.experimental
from tensorflow.experimental import dtensor

#Using tensorflow.keras
from tensorflow.keras import dtensor

有关详细信息，请参阅本指南。 谢谢你。