在Cifar10上更改TensorFlow中的线程数

Question

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

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

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

Answer 1

要配置此值,可以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). (6认同)