美文网首页中华文学
keras如何限定CPU数

keras如何限定CPU数

作者: 读书三万本 | 来源:发表于2020-06-23 21:56 被阅读0次

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

相关文章

网友评论

    本文标题:keras如何限定CPU数

    本文链接:https://www.haomeiwen.com/subject/ogykfktx.html