在 Cifar10 上更改 TensorFlow 中的线程数

Question

每当我运行 cifar10_eval.py 时，都会创建 32 个线程，如下所示：

I tensorflow/core/common_runtime/local_device.cc:25] 本地设备内部操作并行线程：32

我认为这个线程数是运行ning在CPU上的线程数，但是当我查看使用情况时，只有400-500%的CPU被使用。无论如何要改变这个线程数？

Answer 1

要配置这个值，你可以在构造tf.Session时传递一个tf.ConfigProto参数：

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

Changing the number of threads in TensorFlow on Cifar10