python - 如何确定我的 GPU 上的 Tensorflow 的速度？

Question

我刚开始使用Tensorflow（版本 2.3.0）并将其安装在我的 GPU 上（在使用 python 3.5 的虚拟环境中）。这似乎工作正常。我正在使用带有 Windows 10的nvidia geforce 1060。
现在我的问题是：如何找到我的 tensorflow 工作的速度。我发现了这个测试：测试 tensorflow 安装的速度基准，但在我看来，使用我的版本似乎很老了。tensorflow 2有类似的东西吗？我该如何开始呢？我想将它与另一台计算机进行比较，以找出哪台计算机的速度快多少。

干杯

PS：如果我能改进我的帖子，就让我现在。

标签： pythonpython-3.xperformancetensorflowtensorflow2.0

Answer 1

我不知道是否有任何官方的 tensorflow 基准，但在TensorFlow 模型 repo中有一些与 TensorFlow 2 一起使用的官方模型，您可以使用它们来训练（例如 MNIST）并比较两台机器上的训练速度。

或者您可以按照其中一个教程来实现和测试小型网络的速度。