Keras在不使用TensorFlow的情况下从CSV创建数据集

2024-06-14 18:09:56 发布

您现在位置:Python中文网/ 问答频道 /正文

在我发现的每个问题和教程中,tf.data.Dataset都用于CSV文件,但我没有使用tensorflow,我使用的是PlaidML,因为我的AMD GPU在ROCm中不受支持。我尝试过使用相同的代码

os.environ["KERAS_BACKEND"] = "plaidml.keras.backend"
from tensorflow import keras

但这仍然不使用plaidml后端。如何在没有tensorflow的情况下将此数据集https://www.kaggle.com/keplersmachines/kepler-labelled-time-series-data加载到keras中?非常感谢。我通过查看输出来检查是否使用了plaidml后端。如果它说“使用plaidml.keras.backend”,那么它就是在使用plaidml。而且,只有plaidml可以识别我的GPU,因此tensorflow将使用我的CPU


Tags: 文件csv代码backenddatagputftensorflow