web-dev-qa-db-fra.com

Limiter le nombre de cœurs utilisés dans Keras

J'ai une machine partagée avec 64 cœurs sur laquelle j'ai un gros pipeline de fonctions Keras que je veux exécuter. Le fait est qu'il semble que Keras utilise automatiquement tous les cœurs disponibles et je ne peux pas le faire.

J'utilise Python et je veux exécuter 67 réseaux de neurones dans une boucle for. Je voudrais utiliser la moitié des cœurs disponibles.

Je ne trouve aucun moyen de limiter le nombre de cœurs dans Keras ... Avez-vous une idée?

17
Mohamed AL ANI

Comme l'a suggéré @ Yu-Yang, j'ai utilisé cette ligne avant chaque ajustement que je fais:

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)))

Vérifiez l'utilisation du processeur (htop): enter image description here

17
Mohamed AL ANI