我有一个在Windows 10上使用keras和tensorflow(1.15)的python程序。问题是,当在CPU上运行我所构建的卷积网络模型的预测时,默认情况下速度非常慢(0.3-0.4秒) -尽管输入大小非常小(4,3,19,19)。
[当我在Windows资源监视器中检查进程时,我发现python.exe使用了大量线程(〜422)。我有一个相对不错的24核处理器。稍作搜索后,我将OMP_NUM_THREADS环境变量设置为1以查看会发生什么,因此首先set OMP_NUM_THREADS = 1然后启动python。现在,预测时间降到了0.03秒。所以它要快得多。
您知道为什么会这样吗?可以更快地完成吗?
谢谢。
Hai我有你的问题的答案。首先:检查您是否与Keras的master分支保持最新。如果在TensorFlow上运行,请检查您是否是最新版本。如果问题仍然存在,请访问网站。https://www.kdnuggets.com/2019/03/train-keras-model-20x-faster-tpu-free.html(和)https://machinelearningmastery.com/how-to-make-classification-and-regression-predictions-for-deep-learning-models-in-keras/