在Windows 10上使用CPU在Keras顶部运行tensorflow

Question

我已经在Windows 10上的Anaconda安装中安装了Tensorflow和Keras。我使用的是Intel i7处理器。 训练CSV文件的4000个数据样本需要40分钟，我正在尝试对该数据执行LSTM RNN预测分析。

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

Answer 1

是的，在仅使用CPU进行训练时，这似乎是运行代码的合理时间。 如果使用NVIDIA GPU，它将运行得更快。

但是，您可能没有使用CPU上的每个内核。 如果是这样，它可能会运行得更快。 您可以通过运行来更改Tensorflow使用的线程数

sess = tf.Session(config=tf.ConfigProto(intra_op_parallelism_threads=NUM_THREADS))

如果将线程数设置为等于CPU提供的线程数，则它应运行得更快。