报错：cannot import name ‘dtensor‘ from ‘tensorflow.compat.v2.experimental‘

报错：cannot import name 'dtensor' from 'tensorflow.compat.v2.experimental' (/home/ubuntu/miniconda3/envs/tf26/lib/python3.8/site-packages/tensorflow/_api/v2/compat/v2/experimental/init.py)

我的环境

cuda11.2 tensorflow2.6 python 3.8

在网上查了好久，报错是因为keras版本太高，需要降低到和tensorllow版本一致，pip install keras==2.6，降低keras版本就好了。