news 2026/7/23 9:06:13

每个进程分别初始化自己的模型

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
每个进程分别初始化自己的模型

为此我编写了一个python文件来对一个分类模型进行服务化,文件首先进行模型初始化,之后每次web请求,对请求中的数据data利用模型进行预测,返回其对应的标签。

#label_service.py

省略一些引入的包

model = Model() #数据模型
model.load() #模型加载训练好的数据到内存中
app = Flask(name)
class Label(MethodView):
def post(self):
data = request.data
label = model.predict(data)
return label

app.add_url_rule(‘/labelservice/’, view_func=Label.as_view(‘label’), methods=[‘POST’,‘GET’])
利用gunicorn进行启动,gunicorn的好处在于其支持多进程,每个进程可以独立的服务一个外部请求,这样就可以利用多核。

gunicorn -w8 -b0.0.0.0:12711 label_service:app
其中:
-w8 意思是启动8个服务进程。

满心欢喜的启动,但是随即我就发现内存直接爆掉。前面说过,我的模型加载到内存中需要8个G,但是由于我启动了8个工作进程,每个进程都初始化一次模型,这就要求我的机器至少有64G内存,这无法忍受。可是,如果我就开一个进程,那么我的多核机器的CPU就浪费了,怎么办?

那么有没有什么方法能够使得8个工作进程共用一份内存数据模型呢? 很遗憾,python中提供多进程之间共享内存都是对于固定的原生数据类型,而我这里面是一个用户自定义的类。此外,模型中依赖的大量的第三方机器学习包,这些包本身并不支持共享内存方式,而且我也不可能去修改它们的源码。怎么办?

gunicorn 进程模型
仔细看了gunicorn的官方文档,其中就有对其工作模型的描述。

gunicorn主进程:负责fork子进程并监控子进程,根据外部信号来决定是否增加或者减少子进程的数量。
gunicorn子进程:负责接收web请求并且完成请求计算。
我突发奇想,我可以利用gunicorn父子进程在fork时共享父进程内存空间直接使用模型,只要没有对模型的写操作,就不会触发copy-on-write,内存就不会由于子进程数量增加而成本增长。原理图如下:

主进程首先初始化模型,之后fork的子进程直接就拥有父进程的地址空间。接下来的问题就是如何在gunicron的一个恰当的地方进行初始化,并且如何把模型传递给Flask。

实现方式2:利用gunicorn配置文件只在主进程中初始化模型
查看gunicorn官方文档,可以在配置文件配置主进程初始化所需的数据,gunicorn保证配置文件中的数据只在主进程中初始化一次。之后可以利用gunicorn中的HOOK函数pre_request,把model传递给flask处理接口。

#gunicorn.conf
import sys
sys.path.append(“.”) #必须把本地路径添加到path中,否则gunicorn找不到当前目录所包含的类
model = Model()
model.load()
def pre_request(worker, req):
req.headers.append((‘FLASK_MODEL’, model)) #把模型通过request传递给flask。

pre_request = pre_request
#label_service.py

省略一些引入的包

app = Flask(name)
class Label(MethodView):
def post(self):
data = request.data
model = request.environ[‘HTTP_FLASK_MODEL’] #从这里取出模型,注意多了一个HTTP前缀。
label = model.predict(data)
return label

app.add_url_rule(‘/labelservice/’, view_func=Label.as_view(‘label’), methods=[‘POST’,‘GET’])
启动服务:

gunicorn -c gunicorn.conf -w8 -b0.0.0.0:12711 label_service:app
使用 -c 指定我们的配置文件。

启动服务发现达到了我的目的,模型只初始化一次,故总内存消耗还是8G。

这里面提醒大家,当你用top看内存时,发现每个子进程内存大小还是8G,没有关系,我们只要看本机总的剩余内存是减少8G还是减少了8*8=64G。
到此,满心欢喜,进行上线,但是悲剧马上接踵而来。服务运行一段时间,每个进程内存陡增1G,如下图是我指定gunicorn进程数为1的时候,实测发现,如果启动8个gunicorn工作进程,则内存在某一时刻增长8G,直接oom。

到此,我的内心是崩溃的。不过根据经验我推测,在某个时刻某些东西触发了copy-on-write机制,于是我让研究院小伙伴仔细审查了一下他们的模型代码,确认没有写操作,那么就只可能是gunicorn中有写操作。

接下来我用蹩脚的英文在gunicorn中提了一个issue:https://github.com/benoitc/gunicorn/issues/1892 ,大神立刻给我指出了一条明路,原来是python的垃圾收集器搞的鬼,详见:https://bugs.python.org/issue31558 , 因为python的垃圾收集会更改每个类的 PyGC_Head,从而它触发了copy-on-write机制,导致我的服务内存成倍增长。

那么有没有什么方法能够禁止垃圾收集器收集这些初始化好的需要大内存的模型呢?有,那就是使用gc.freeze(), 详见 https://docs.python.org/3.7/library/gc.html#gc.freeze 。但是这个接口在python3.7中才提供,为此我不得不把我的服务升级到python3.7。

实现方式3:python2.7升级到python3.7后使用gc.freeze()
升级python是一件非常痛苦的事情,因为我们的代码都是基于python2.7编写,许多语法在python3.7中不兼容,特别是字符串操作,简直恶心到死,只能一一改正,除此之外还有pickle的不兼容等等,具体修改过程不赘述。最终我们的服务代码如下。

#gunicorn.conf
import sys
import gc
sys.path.append(“.”) #必须把本地路径添加到path中,否则gunicorn找不到当前目录所包含的类
model = Model()
model.load()
gc.freeze() #调用gc.freeze()必须在fork子进程之前,在gunicorn的这个地方调用正好合适,freeze把截止到当前的所有对象放入持久化区域,不进行回收,从而model占用的内存不会被copy-on-write。
def pre_request(worker, req):
req.headers.append((‘FLASK_MODEL’, model)) #把模型通过request传递给flask。

pre_request = pre_request
上线之后观察到,我们单个进程内存大小从8个G降低到6.5个G,这个推测和python3.7本身的优化有关。其次,运行一段时间后,每个子进程内存缓慢上涨500M左右后达到稳定,这要比每个子进程突然增加1G内存(并且不知道是否只突增一次)要好的多。

使用父子进程共享数据后需要进行预热
当使用gunicorn多进程实现子进程与父进程共享模型数据后,发现了一个问题:就是每个子进程模型的第一次请求计算耗时特别长,之后的计算就会非常快。这个现象在每个进程拥有自己的独立的数据模型时是不存在的,不知道是否和python的某些机制有关,有哪位小伙伴了解可以留言给我。对于这种情况,解决办法是在服务启动后预热,人为尽可能多发几个预热请求,这样每个子进程都能够进行第一次计算,请求处理完毕后再上线,这样就避免线上调用方长时间hang住得不到响应

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/23 9:05:00

慢病毒载体伯远生物慢病毒载体

慢病毒载体伯远生物慢病毒载体 伯远生物是国家级专精特新小巨人企业,国家级重点实验室,牵头多项省部级重大专项,公司科研技术人员500(硕博占比40%以上),作为功能基因研究综合性平台, 15年技术沉…

作者头像 李华
网站建设 2026/7/23 9:03:09

LabVIEW与Node-RED通过MQTT实现工业物联网通信实战

在实际工业自动化和物联网项目中,经常需要将 LabVIEW 这类强大的数据采集与控制工具,与 Node-RED 这类轻量级流程编排工具打通。MQTT 协议凭借其轻量、异步、解耦的特性,成为连接两者的理想桥梁。但很多工程师在具体实施时,会遇到…

作者头像 李华
网站建设 2026/7/23 9:00:45

GLM5模型性能优化实战:从计算图到硬件加速

1. GLM5模型重构性能优化全攻略:从理论到实践最近在重构一个基于GLM5的推荐系统模型时,发现原有实现存在严重的性能瓶颈。经过两周的调优,最终将推理速度提升了3.2倍,内存占用降低了58%。今天就把这次性能优化的完整思路和实操经验…

作者头像 李华
网站建设 2026/7/23 9:00:15

GPU运维与大模型微调实战:LLaMA-Factory应用指南

1. GPU运维与大模型微调概述 在当今AI技术快速发展的背景下,GPU已成为大模型训练和推理的核心硬件基础。LLaMA-Factory作为开源的大模型微调框架,极大降低了开发者进行模型定制化的门槛。这套工具链特别适合在AutoDL等云GPU平台上运行,能够高…

作者头像 李华
网站建设 2026/7/23 8:58:47

高德地图9.5Beta版评测:主题美化与导航优化

1. 高德9.5Beta版本初体验:界面与交互升级 作为一名长期使用高德地图的深度用户,这次9.5Beta版本的更新确实带来了不少惊喜。安装包体积控制在85MB左右,相比前代版本优化明显。首次启动时会有一个简洁的过渡动画,随后进入重新设计…

作者头像 李华
网站建设 2026/7/23 8:57:25

华硕W419L笔记本升级SSD与内存实战指南

1. 华硕W419L老笔记本升级方案概述 2015年上市的华硕W419L作为一款经典商务本,采用第四代Intel酷睿处理器搭配4GB DDR3内存和500GB机械硬盘的配置。七年过去,这套配置在运行现代办公软件时已明显力不从心——开机耗时超过1分钟,PS打开10MB图片…

作者头像 李华