Tensorflow: Allocation of XXXXXX exceeds 10% of system memory 可能的原因

在使用他人的项目代码时,遇到了这个问题

W T:\src\github\tensorflow\tensorflow\core\framework\allocator.cc:108]:Allocation of XXXXXX exceeds 10% of system memory

意思是方法分配的空间的大小超过了系统内存的10%

配置:Tensorflow 1.10.0,内存 128G

这算是一种Warning,而不是Error

在客观上无法马上扩展内存的情况下,仿佛只能通过减少数据量和模型参数量的角度进行修改

综合已有关于此问题的一些讨论[1][2],可能的角度如下,但不适用于所有可能的情况:

0、尽可能关闭其他占用内存较多的不必要的程序

1、减小batch_size

2、减少数据量,比如处理图片时先进行压缩

3、减少模型参数,比如层数和节点数

4、增加项目所在硬盘的虚拟内存交换分区的大小(swap file)

使用交换文件将内存作为 Amazon EC2 实例中的交换空间分配

5、指定物理设备是否内存生长,避免初始化时分配所有内存

tf.config.experimental.set_memory_growth  |  TensorFlow v2.11.0

6、改变优化函数,比如Adam会使用分配模型两倍的参数

7、避免使用大精度的数据类型,比如使用float32而不是float64

愿天堂没有bug

[1] python - Tensorflow Allocation Memory: Allocation of 38535168 exceeds 10% of system memory - Stack Overflow

[2] tensorflow/core/framework/allocator.cc:101] Allocation of X exceeds 10% of system memory. #18735 · Issue #18736 · tensorflow/tensorflow · GitHub

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

冰糖不在家

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值