python - 当整个图在 GPU 上运行时，Tensorflow CPU 使用率 > 100%

Question

我有简单的张量流代码：

import tensorflow as tf
from tensorflow.python.client import timeline

zeros = tf.zeros(shape=(1000, 1000))
a, b = tf.while_loop(
    lambda a, b: a < 1000,
    lambda a, b: (a + 1, tf.matmul(zeros, zeros)),
    (0, zeros))

with tf.Session() as sess:
    options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
    run_metadata = tf.RunMetadata()
    sess.run([a,b], options=options, run_metadata=run_metadata)

    fetched_timeline = timeline.Timeline(run_metadata.step_stats)
    chrome_trace = fetched_timeline.generate_chrome_trace_format()

    with open('timeline.json', 'w') as f:
        f.write(chrome_trace)

我不明白，为什么要top显示 >100 %cpu，因为所有操作都在 gpu 上运行：trace_data

有任何想法吗？

标签： pythontensorflowcpu-usage