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

156

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

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

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

спросил(а) 5 лет, 2 месяца назад

Zk1001

пожаловаться

Решение

204

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

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

ответил(а) 5 лет, 2 месяца назад

mrry

пожаловаться

Ваш ответ

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

Еще в рубрике

Другие вопросы