当前位置:  开发笔记 > 开发工具 > 正文

在Cifar10上更改TensorFlow中的线程数

如何解决《在Cifar10上更改TensorFlow中的线程数》经验,为你挑选了1个好方法。

每当我运行cifar10_eval.py时,在创建32个线程时如下:

I tensorflow/core/common_runtime/local_device.cc:25]本地设备intra op parallelism threads:32

我认为这个线程数是在CPU上运行的线程数,但是当我检查使用情况时,只使用了400-500%的CPU.反正有没有改变这个线程数?



1> mrry..:

要配置此值,可以tf.ConfigProto在构造时传递参数tf.Session:

NUM_THREADS = …
sess = tf.Session(config=tf.ConfigProto(
    intra_op_parallelism_threads=NUM_THREADS))


线程是操作线程池的标准pthread,由[此文件]中的C++代码创建(https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/common_runtime/direct_session.cc).
推荐阅读
凹凸曼00威威_694
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有