为什么要限定CPU数
- 你和小伙伴共用资源,你一下子把资源沾满了,大家只能默默地把你的程序kill掉,跑了一半的程序,停了,就问你是否心碎?也不能全怪大家,谁叫你不顾大家死活?你要出结果,人家也要啊!
- 你可能需要同时进行多路实验,那么资源有限的情况下?你只能把蛋糕(CPU资源)分成很好多份了!
方法
from keras import backend as K;
主程序添加如下代码,把5改成你想要的
K.set_session(K.tf.Session(config=K.tf.ConfigProto(intra_op_parallelism_threads=5, inter_op_parallelism_threads=5))