运行 Keras 之上的 tensorflow,CPU 在 Windows 10

Running tensorflow on top of Keras with CPU on Windows 10

我在 Windows10 上安装了 Tensorflow 和 Keras,并安装了 Anaconda。我使用的是 Intel i7 处理器。训练一个 CSV 文件的 4000 个数据样本需要 40 分钟,我正在尝试对此数据执行 LSTM RNN 预测分析。

这是使用 CPU 的预期编译时间吗?我们可以使用 cpu 或切换到 GPU 使其更快吗?

是的,当您仅使用 CPU 进行训练时,这似乎是您的代码达到 运行 的合理时间。如果您使用 NVIDIA GPU,它会 运行 快得多。

但是,您可能不会使用 CPU 上的每个内核;如果是,它可能 运行 更快。您可以通过 运行ning

更改 Tensorflow 使用的线程数
sess = tf.Session(config=tf.ConfigProto(intra_op_parallelism_threads=NUM_THREADS))

如果您将线程数设置为等于 CPU 提供的线程数,它应该 运行 更快。