4

How can I profile TensorFlow models? I want at least to find ways to make the model faster, see how much GPU load I have, and how much GPU memory is consumed.

Jekyll SONG
  • 479
  • 3
  • 6
  • 18
Konstantin Solomatov
  • 10,252
  • 8
  • 58
  • 88
  • Possible duplicate of [Can I measure the execution time of individual operations with TensorFlow?](https://stackoverflow.com/questions/34293714/can-i-measure-the-execution-time-of-individual-operations-with-tensorflow) – benjaminplanche Apr 05 '18 at 08:44

1 Answers1

6

There are two ways to profile models. One way is a tensorboard. Here is a comprehensive tutorial about it and here is a good video.

Additionally, clicking on a node will display the exact total memory, compute time, and tensor output sizes.

enter image description here

Another way is tensorflow debugger, which also has tutorials.

Salvador Dali
  • 214,103
  • 147
  • 703
  • 753