首页 > 解决方案 > 在 Tensorflow 中,第一批的 GPU 处理时间比其他任何时间都长得多

问题描述

我想知道每个batch在GPU单元中是如何处理的,所以我输出了每个batch的开始和结束时间。但是第一个batch的处理时间比其他batch要长得多。这是我获取每个批次的处理时间的代码 在此处输入图像描述

这是 GPU 中前四批的处理时间 在此处输入图像描述

为了比较,我还输出了 CPU 中的处理时间在此处输入图像描述

我们可以发现,虽然 CPU 的总时间比 GPU 长,但第一批的处理时间更短。

标签: tensorflow

解决方案


推荐阅读