OMP: Error #34: System unable to allocate necessary resources for OMP thread:

本文详细介绍了在使用TensorFlow进行模型训练时遇到的OMP线程过多错误,并提供了一种有效的解决方案,通过调整TensorFlow配置参数来限制线程数,避免系统资源分配错误。
部署运行你感兴趣的模型镜像

在使用TensorFlow训练模型时,突然报错出现OMP线程过多的问题,网上查找了一下资料可以减少线程数,下面是报错的具体内容:

INFO:tensorflow:global step 80: loss = 2.3479 (3.266 sec/step)
INFO:tensorflow:global step 84: loss = 2.2638 (1.569 sec/step)
OMP: Error #34: System unable to allocate necessary resources for OMP thread:
OMP: System error #11: Resource temporarily unavailable
OMP: Hint: Try decreasing the value of OMP_NUM_THREADS.
Aborted (core dumped) 

网上有人说通过设置环境变量参数改变线程数:

os.environ["OMP_NUM_THREADS"] = "12"
os.environ["KMP_BLOCKTIME"] = "1"
os.environ["KMP_AFFINITY"] = "granularity=fine,compact,1,0"
os.environ["KMP_SETTINGS"] = "0"

但是这种办法并不能解决这个问题

解决办法:

config = tf.ConfigProto()
config.intra_op_parallelism_threads = 12
config.inter_op_parallelism_threads = 2
tf.session(config=config)

 

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值