tensorflowGPU優化之利用timeline詳細查看各部分op時間

在利用tensorflow寫程序是,我們常常會碰到GPU利用率始終不高的情況,這時我們需要詳細瞭解程序結點的消耗時間,tensorboard提供了一個窗口,但仍不詳細,這裏介紹timeline的使用,他可以更詳細的給出各部分op的時間消耗,讓你瞭解程序的瓶頸 簡單的例子: import tensorflow as tf from tensorflow.python.client import ti
相關文章
相關標籤/搜索