linux环境下经常内存溢出的一种解决办法

本文讲述了在Linux环境下,遇到内存溢出问题并非总是由batch_size过大引起,而是可能因为模型未正确加载。通过将本地正确模型打包上传至Linux环境并解压,成功解决了程序内存溢出的问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

首先说一下,内存经常溢出,爆内存不一定是因为batch_size过大,也有可能是模型没有加载上。弄了一天才把这个问题解决

问题:
我在使用paddlehub的下面这段函数时,经常爆内存

paddlehub.Module(name='deeplabv3p_xception65_humanseg')

简单说一下过程

 //我先用humanseg加载模型,这个deeplabv3p_xception65_humanseg
 //模型如果Linux环境未安装,则自动安装
 humanseg = paddlehub.Module(name='deeplabv3p_xception65_humanseg')
 //开始遍历frames下面的所有图片
 files = [frames + i for i in os.listdir(frames)]
 //开始抠图
 humanseg.segmentation(data={'image': files})  # 抠图

其实在这里,我就开始内存飙升,而且程序直接被killed掉

 humanseg = paddlehub.Module(name='deeplabv3p_xception65_humanseg')

我就当时在想是不是batch_size是不是过大。这时候我就又开始只扣一张图片,不是扣一个视频。可是依然不行,我这时候就思考是不是.paddlehub下面的deeplabv3p_xception65_humanseg模型没有加载成功。
这时候我就把我电脑里这个模型打包成zip上传到linux环境下,然后解压到指定

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值