У меня есть общая машина с 64 ядрами, на которой у меня есть большой конвейер функций Keras, которые я хочу запустить. Дело в том, что кажется, что Keras автоматически использует все доступные ядра, и я не могу этого сделать.
Я использую Python, и я хочу запустить 67 нейронных сетей в цикле for. Я хотел бы использовать половину доступных ядер.
Я не могу найти способ ограничить количество ядер в Keras... У вас есть ключ?
Как предположил @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): 