如何在GPU上确定Tensorflow的速度



我刚开始使用Tensorflow(2.3.0版(,并将其安装在我的GPU上(在使用python 3.5的虚拟环境中(。这似乎很好。我正在使用nvidia geforce 1060和windows 10。
现在我的问题是:如何找到tensorflow的工作速度。我发现了这个测试:测试tensorflow安装的速度基准测试,但在我看来,它对我的版本来说已经过时了。tensorflow 2有类似的东西吗?我该怎么开始呢?我想把它和另一台电脑比较一下,看看哪台电脑快多少。

干杯

附言:如果我能改进我的帖子,现在就让我吧。

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

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

我在另一篇之前没有找到的帖子中找到了完整的答案:

对tensorflow图的部分进行基准测试的正确方法是什么?

最新更新