TensorFlow作业默认使用多个核心吗?

时间:2017-03-16 21:40:06

标签: tensorflow

我正在使用TensorFlow模型repository运行imagenet。我已按照Github comment中的说明对sess.run进行了检测,并在chrome:// tracing

中获得了以下视图

enter image description here

我想知道TF有时会一直使用多核或单核。当操作可以并行运行时,我认为它正在使用多个核心,如图中的红色框所示。但是,所有这6个线程都列在/ job:localhost / replicate:0 / task:0 / cpu:0下,这让我对我的解释提出质疑。 cpu:0是指所有CPU内核吗?

我在8核的桌面上运行。我运行htop以查看TF运行期间的核心利用率,我看到只有一个核心达到饱和95-100%。

1 个答案:

答案 0 :(得分:19)

我发现了existing answer这个问题。所有内核都包装在cpu:0中,即TensorFlow确实默认使用多个CPU内核。