Изменение количества потоков в TensorFlow на Cifar10

Всякий раз, когда я запускаю cifar10_eval.py, создается 32 потока следующим образом:

я tensorflow/core/common_runtime/local_device.cc: 25] Локальное устройство int op parallelism threads: 32

Я думаю, что это число потоков - это количество потоков, работающих на процессорах, но когда я проверяю использование, используется только 400-500% процессоров. Есть ли способ изменить это число потоков?

Ответы

Ответ 1

Чтобы настроить это значение, вы можете передать аргумент tf.ConfigProto при построении tf.Session:

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