tensorflow - 如何在 GPU 上运行 tensorflow 基准测试工具？

Question

如何在 GPU 上运行 tensorflow 基准测试工具？ https://github.com/tensorflow/tensorflow/tree/master/tensorflow/tools/benchmark

Answer 1

Tensorflow 包含一个抽象类，为 TensorFlow 基准测试提供帮助：Benchmark

因此，可以制作 Benchmark 对象并用于在 tensorflow 图的一部分上执行基准测试。在下面的代码中，实例化了一个基准对象，然后run_op_benchmark调用该方法。

在run_op_benchmark会话中传递，conv_block 张量（在这种情况下），feed_dict，烧伤迭代次数，所需的最小迭代次数，布尔标志，以防止基准也计算内存使用和方便的名称。

该方法返回一个包含基准测试结果的字典：

benchmark = tf.test.Benchmark()
results = benchmark.run_op_benchmark(sess=sess, op_or_tensor=z_tf, 
                                     feed_dict={x_tf: x_np}, burn_iters=2, 
                                     min_iters=n_iter, 
                                     store_memory_usage=False, name='example')

有关在 TensorFlow-GPU v2.3 上运行的代码片段，请参阅我的答案https://stackoverflow.com/a/63591009/2478346