【无标题】

Tensorflow无法导入DTensor错误

  • 问题描述
    • 问题原因
    • 解决方法

问题描述

在使用tensorflow调用tensorflow.keras.datasets.cifar10.load_data导入cifar10数据集时报错:

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

问题原因

考虑是keras、tensorflow.keras、tensorflow.python.keras之间的冲突导致

解决方法

from tensorflow.keras.datasets.cifar10 import load_data

替换为

from tensorflow.python.keras.datasets.cifar10 import load_data

你可能感兴趣的:(机器学习,tensorflow,keras,深度学习)