Kerasで使用されるコアの数を制限する
64コアの共有マシンで、実行したいKeras関数の大きなパイプラインがあります。問題は、Kerasが利用可能なすべてのコアを自動的に使用しているように見えることであり、それはできません。
Pythonを使用し、forループで67個のニューラルネットワークを実行したい。使用可能なコアの半分を使用したい。
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)))
CPU使用率（htop）を確認します： 