Question

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

Answer 1

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

因此，可以制作一个Benchmark对象并将其用于在张量流图的一部分上执行基准。在下面的代码中，将实例化一个基准对象，然后调用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

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

1 个答案: