keras如何限定CPU数

为什么要限定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))

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容