Search code examples
tensorflowtensorboard

How to display per-node memory in TensorBoard?

In another question, someone showed a screenshot of his TensorBoard, displaying memory usage per node:

enter image description here

I never see those in my experiments with Tensorboard. All I'm doing is calling

writer = tf.summary.FileWriter('/tmp/tensorboard', sess.graph)

after sess.run(). Are there perhaps some "summaries" that I need to add to record memory usage?

Solution

  • Following the instruction here, you might actually have written all necessarry lines for memory and time to be displayed, but you also need to change the "Sessions runs" drop-down in the tensorboard GUI from default "None" to any of the options. Otherwise they are hidden. See graphic:

    enter image description here

    At least, this is what I needed to do to display them - the previous answers worked for me, but I needed to change the drop-down.