cannot import name ‘dtensor’ from ‘tensorflow.compat.v2

今天运行tf测试代码出现错误 cannot import name 'dtensor' from 'tensorflow.compat.v2，查了下发现是我安装的Tensorflow 和 Keras版本不一致导致的。

pip list | grep  tensorflow 是 2.6.0的，而 keras是 2
pip list | grep ke rs.9.0的，实际我只安装了

发现 tensorflow 是 2.6.0的，而 keras 是2.9.0的，实际我只安装了 tensorflow ，keras 是当作 tensorflow 依赖被安装上的，默认安装了最新版本。

解决方法：

pip install keras==2.6.0

安装对应版本的 keras 即可。