开发者社区> 问答> 正文

跑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。。

很是让人奇怪~~~ 这个该怎么解决呢?

本问题来自阿里云开发者社区的【11大垂直技术领域开发者社群】。 https://developer.aliyun.com/article/706511 点击链接欢迎加入感兴趣的技术领域群

展开
收起
茶什i 2019-09-06 14:54:15 955 0
0 条回答
写回答
取消 提交回答
问答排行榜
最热
最新

相关电子书

更多
RDS SQL Server CPU高使用率性能优化 立即下载
十分钟上线-使用函数计算构建支付宝小程序服务 立即下载
低代码开发师(初级)实战教程 立即下载