ImportError: cannot import name 'dtensor' from 'tensorflow.compat.v2.experimental'

from tensorflow.compat.v2.experimental import dtensor as dtensor_api  # pylint: disable=g-import-not-at-top
ImportError: cannot import name 'dtensor' from 'tensorflow.compat.v2.experimental' (/home/user/venv/lib/python3.8/site-packages/tensorflow/_api/v2/compat/v2/experimental/__init__.py)

• tensorflow와 keras 버전 비호환으로 생기는 에러
• tensorflow-gpu==2.6.0
• keras==2.9

해결방법

• pip install keras==2.6
• keras나 tensorflow 둘 중하나를 버전에 맞게 맞춰준다.