在Tensorflow的2.1或2.2稳定版本中，缺少`keras.preprocessing.image_dataset_from_directory()`函数。为解决此问题，你可以通过`pip install tf-nightly`安装开发版本。然而，这样做的副作用是安装夜间版本后，GPU支持与`image_dataset_from_directory()`不兼容，反之亦然。需权衡选择稳定版与特定功能的使用。
出现这种问题是由于2.1or2.2稳定版本的tensorflow没有这个函数：
train_dst = keras.preprocessing.image_dataset_from_directory(）
如果不想更换版本的话可以利用下面方式进行解决：
pip install tf-nightly
注：安装tf-nightly后，GPU能用tf.keras.preprocessing.image_dataset_from_directory就不能用，反之，tf.keras.preprocessing.image_dataset_from_directory能用，GPU就不能用了
您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型
1万+
2792
1万+
1万+
2744
2万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
