Изменение количества потоков в TensorFlow на Cifar10
156
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
5 лет, 2 месяца назад
добавить комментарий
пожаловаться
Решение
204
Чтобы настроить это значение, вы можете передать аргумент 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
5 лет, 2 месяца назад
добавить комментарий
пожаловаться
Ваш ответ
- Вопросы
- Tensorflow
- Изменение количества потоков в TensorFlow на Cifar10