导入错误：无法从"tensorflow.compat.v2.experimental"导入名称"dtensor"

Question

我正在尝试使用tensorflow作为分配创建对象检测。

ImportError: cannot import name 'dtensor' from 'tensorflow.compat.v2.experimental' (c:Project.venvlibsite-packagestensorflow_apiv2compatv2experimental__init__.py)我将感激任何帮助!

Answer 1

我得到了同样的错误。我降级到Keras v2.6.0，它修复了这个问题。因此，在16.05.2022，这似乎是由v2.9中的一个错误引起的。*

要降级到v2.6.0，运行:

!pip install keras==2.6.*

Answer 2

请再次尝试将TensorFlow版本升级到最新版本2.9并重新启动内核(运行时)以解决此问题。

!pip install tensorflow==2.9
#Or
!pip install --quiet --upgrade --pre tensorflow

例如:

import tensorflow as tf
from tensorflow.experimental import dtensor
print('TensorFlow version:', tf.__version__)
mesh_1d = dtensor.create_mesh([('x', 6)])#, devices=DEVICES)
print(mesh_1d)

输出:

TensorFlow version: 2.9.1
<tensorflow.dtensor.python.layout.Mesh object at 0x7f890d7ce690>

请点击此链接了解更多详情。