lmw0320 2019-09-06 09:16 采纳率: 75%
浏览 1188
已采纳

跑keras模型,设置CPU使用上限报错是怎么回事

跑个keras算法库的模型,为防止电脑被占用过多资源,我设置了CPU的使用内核数,具体代码如下:

import tensorflow as tf
import keras.backend.tensorflow_backend as KTF

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' #忽略CPU编译不支持警告
config = tf.ConfigProto()
config.device_count = {'CPU': 4} #该句代码注释掉,才不会报错
config.intra_op_parallelism_threads = 4
config.inter_op_parallelism_threads = 4
config.allow_soft_placement = True
config.log_device_placement = True #不打印设备分配日志
sess = tf.Session(config=config)
KTF.set_session(sess)

运行发现报错:Assignment not allowed to repeated field "device_count" in protocol message object。。
很是让人奇怪~~~
这个该怎么解决呢?

  • 写回答

2条回答 默认 最新

  • lmw0320 2019-09-11 08:16
    关注

    从简书上请教了别人,应该是设置为config.device_count ['CPU']= 4,就发现没有报错。。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

悬赏问题

  • ¥15 Mapreduce是正常的,在运行其他jar包时并没有任何问题,只是在做LogCount.jar 时出的问题。如图所示
  • ¥15 ImportError: DLL load failed while importing _iterative: 找不到指定的模块。
  • ¥15 如何通过交互分析得出某高危患者对放疗获益更多
  • ¥15 相关性分析中,p<0.05, r=0.29,怎么评价相关性呢
  • ¥15 docker部署Mongodb后输入命令报错?
  • ¥15 将下列流程图转变成python程序代码
  • ¥15 我需要全国每个城市的最新小区名字等数据。
  • ¥15 开发一个小区生态的小程序
  • ¥15 如何解决Excel中dependent dropdown list 的问题
  • ¥15 MddBootstrapInitialize2失败