Limit number of cores used in Keras

前端 未结 2 1290
北恋
北恋 2020-12-08 20:14

I have a shared machine with 64 cores on which I have a big pipeline of Keras functions that I want to run. The thing is that it seems that Keras automatically uses all the

2条回答
  •  清歌不尽
    2020-12-08 20:48

    As @Yu-Yang suggested, I used this line before each fit I do :

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

    Check the CPU usage (htop) :

提交回复
热议问题