Ограниченное количество ядер, используемых в Keras

У меня есть общая машина с 64 ядрами, на которой у меня есть большой конвейер функций Keras, которые я хочу запустить. Дело в том, что кажется, что Keras автоматически использует все доступные ядра, и я не могу этого сделать.

Я использую Python, и я хочу запустить 67 нейронных сетей в цикле for. Я хотел бы использовать половину доступных ядер.

Я не могу найти способ ограничить количество ядер в Keras... У вас есть ключ?

Ответ 1

Как предположил @Yu-Yang, я использовал эту строку перед каждым положением, которое я делаю:

from keras import backend as K
K.set_session(K.tf.Session(config=K.tf.ConfigProto(intra_op_‌​parallelism_threads=‌​32, inter_op_parallelism_threads=32)))

Проверьте использование ЦП (htop): enter image description here