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

Nod*_*rov 14 tensorflow

我正在从TensorFlow模型运行imagenet .我已经在Github评论中描述了sess.run,并在chrome:// tracing中获得了以下视图

在此输入图像描述

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

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

Nod*_*rov 21

我找到了这个问题的现有答案.所有内核都包装在cpu:0中,即TensorFlow默认情况下确实使用多个CPU内核.


归档时间:

查看次数:

17389 次

最近记录:

8 年,5 月 前