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

152
25

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


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


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

спросил(а) 2015-12-21T09:19:00+03:00 4 года, 11 месяцев назад
1
Решение
197

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

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

ответил(а) 2015-12-21T10:09:00+03:00 4 года, 11 месяцев назад
Ваш ответ
Введите минимум 50 символов
Чтобы , пожалуйста,
Выберите тему жалобы:

Другая проблема