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

有關詳細信息，請參閱本指南。 謝謝你。