gunicorn+flask使用问题修复方案
admin
2024-05-22 20:51:17
0

来源:torch模型在线推理采用gunicorn+flask部署API服务

问题简述:torch模型——huggingface bert模型,CPU核数96+4个GPU同时使用,启动worker=80左右(80个并发请求),前期能够调用GPU,后期发现有些进程重启后就不再被使用了(top下看不到该pid的进程),因此使用GPU的就不再被使用了,只剩下使用CPU的部分worker,请求方描述,只改变了 java中try请求等待时间加长了。测试也发现postman单个请求响应时间为8min,是以前的至少4倍(之前最多2min,基本上在1min内响应)。

当全部只采用GPU训练(随机选择device),采用kill -9杀死pid时,ps下面还有新的进程产生,这是咋回事?经搜索发现无法杀死gunicorn子进程,它会再次拉起新的进程(这种进程都是死的,根本无法执行,当有请求过来时返回500),因此如果想要每个GPU启动4-5个进程,只能选择nginx转发,指定单个GPU 4~5个works,但nginx转发可能存在网关耗时问题(暂不考虑,另外nginx怎么和我之前用的不太一样了啊,yum install nginx安装的,之前是ubuntn结果)

为啥要杀死某个进程?这是因为启动的时候无法选定所有GPU均匀分布,也就是说做不到4个卡每个卡4个进程(采用random随机选择GPU index,只有一个脚本的情况下必然无法指定GPU,而如果采用preload_app=False,结果只会选定某个GPU),因此设定preload_app=True,它会共享某部分数据(内存),而模型的数据(输入输出)并没有共享。如果某个卡进程数(pid个数)较少,某个卡较多,这肯定不太好,不能充分利用GPU效率。

为啥出现GPU不能用的情况?不被利用的情况,在之前启动时用了GPU(杀死了部分不合理的pid,该卡进程较多,所以杀了几个)和全部CPU,而CPU进程占比较多(80%+),如果仅仅是因为阻塞导致重启该进程,那么CPU进程最后还能运行(可能有些CPU进程就是没死,部分CPU进程没有发生阻塞)而GPU却死了不能重启。无解,只能增加GPU数,或者多台机器部署。

在gunicorn的执行config中,设置的worker数应该与random选择的index相适合,比如全部用GPU情况下index=[0,1,2,3],那么worker应设置为4的倍数,比如8,16,20,但index如果仅仅为[0,1,2,3],可能会出现某个卡分配的进程多,有的分配的少,为了足够随机,index应该设置为更大长度,比如,[0,1,2,3]*20或者[0123]*40 .这种设置会足够随机(random使用)。

目前worker_class仍旧是采用gevent模式,sync未尝试,因为模型推理可能较慢,异步模式好一点吧。

参考文档:flask+gunicorn,flask加载两次?gunicorn启动 ,nginx问题,nginx部署

相关内容

热门资讯

linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
C++ 机房预约系统(六):学... 8、 学生模块 8.1 学生子菜单、登录和注销 实现步骤: 在Student.cpp的...
A.机器学习入门算法(三):基... 机器学习算法(三):K近邻(k-nearest neigh...
数字温湿度传感器DHT11模块... 模块实例https://blog.csdn.net/qq_38393591/article/deta...
有限元三角形单元的等效节点力 文章目录前言一、重新复习一下有限元三角形单元的理论1、三角形单元的形函数(Nÿ...
Redis 所有支持的数据结构... Redis 是一种开源的基于键值对存储的 NoSQL 数据库,支持多种数据结构。以下是...
win下pytorch安装—c... 安装目录一、cuda安装1.1、cuda版本选择1.2、下载安装二、cudnn安装三、pytorch...
MySQL基础-多表查询 文章目录MySQL基础-多表查询一、案例及引入1、基础概念2、笛卡尔积的理解二、多表查询的分类1、等...
keil调试专题篇 调试的前提是需要连接调试器比如STLINK。 然后点击菜单或者快捷图标均可进入调试模式。 如果前面...
MATLAB | 全网最详细网... 一篇超超超长,超超超全面网络图绘制教程,本篇基本能讲清楚所有绘制要点&#...
IHome主页 - 让你的浏览... 随着互联网的发展,人们越来越离不开浏览器了。每天上班、学习、娱乐,浏览器...
TCP 协议 一、TCP 协议概念 TCP即传输控制协议(Transmission Control ...
营业执照的经营范围有哪些 营业执照的经营范围有哪些 经营范围是指企业可以从事的生产经营与服务项目,是进行公司注册...
C++ 可变体(variant... 一、可变体(variant) 基础用法 Union的问题: 无法知道当前使用的类型是什...
血压计语音芯片,电子医疗设备声... 语音电子血压计是带有语音提示功能的电子血压计,测量前至测量结果全程语音播报࿰...
MySQL OCP888题解0... 文章目录1、原题1.1、英文原题1.2、答案2、题目解析2.1、题干解析2.2、选项解析3、知识点3...
【2023-Pytorch-检... (肆十二想说的一些话)Yolo这个系列我们已经更新了大概一年的时间,现在基本的流程也走走通了,包含数...
实战项目:保险行业用户分类 这里写目录标题1、项目介绍1.1 行业背景1.2 数据介绍2、代码实现导入数据探索数据处理列标签名异...
记录--我在前端干工地(thr... 这里给大家分享我在网上总结出来的一些知识,希望对大家有所帮助 前段时间接触了Th...
43 openEuler搭建A... 文章目录43 openEuler搭建Apache服务器-配置文件说明和管理模块43.1 配置文件说明...