如何在 GPU 上运行张量流基准测试工具

Question

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

Answer 1

Tensorflow 包含一个抽象类，为 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