【问题标题】:ImportError: cannot import name 'dtensor' from 'tensorflow.compat.v2.experimental' import KerasImportError: cannot import name \'dtensor\' from \'tensorflow.compat.v2.experimental\' import Keras
【发布时间】:2022-08-22 19:34:45
【问题描述】:
导入 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
有谁知道如何解决这个错误?
标签:
python
tensorflow
keras
【解决方案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
如需更多信息,请参阅此guide。谢谢你。