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

Question

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

- Misaya.Z

---

你是在寻找类似这样的东西吗？？https://github.com/soumith/convnet-benchmarks/tree/master/tensorflow - Harsha Pokkalla

不，我想要知道每个操作的计算时间 @hars - Misaya.Z

Answer 1

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

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

run_op_benchmark方法传入session、conv_block Tensor（在本例中），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://dev59.com/kbroa4cB1Zd3GeqPiFRa#63591009中的回复。

- CATALUNA84