0

0

Python 在线推理的延迟优化

舞夢輝影

舞夢輝影

发布时间:2026-02-24 21:09:04

|

100人浏览过

|

来源于php中文网

原创

本地部署需切断网络依赖,提前下载模型并设local_files_only=true;预编译模型、启用cudnn.benchmark、避免重复设备转换、用inference_mode、优化io与序列化、异步数据搬运、关闭冗余padding,并手动打点定位瓶颈。

python 在线推理的延迟优化

模型加载阶段卡顿严重

冷启动时 torch.loadtransformers.AutoModel.from_pretrained 耗时几十秒,不是显存不足,而是默认从 Hugging Face Hub 拉权重。本地部署必须切断网络依赖。

  • 把模型用 snapshot_download 提前拉到本地,路径传给 from_pretrained,别让它现场下载
  • local_files_only=True 参数,防止意外回源;同时检查 ~/.cache/huggingface/transformers 是否有残留 symlink
  • torch.jit.tracetorch.compile(PyTorch 2.0+)预编译模型,首次推理仍慢,但后续稳定快 15–30%

单次推理耗时波动大(尤其 batch_size=1)

GPU 利用率低、TensorRT 不生效、CUDA stream 空转——本质是没让计算流水线跑起来。

  • 确认是否启用了 torch.backends.cudnn.benchmark=True,它对固定 shape 输入有加速,但首次运行会多花几毫秒测 kernel
  • 避免在推理循环里反复调用 .to('cuda').half(),提前做好 device + dtype 转换
  • batch_size=1 时,torch.inference_mode()torch.no_grad() 更轻量,显存占用略低,延迟更稳

HTTP 接口层拖慢端到端延迟

FastAPIFlask 包一层后,P99 延迟翻倍,问题常出在 JSON 序列化和同步 IO 上。

AmEav WebSite 企业网站管理系统1.0
AmEav WebSite 企业网站管理系统1.0

系统功能强大、操作便捷并具有高度延续开发的内容与知识管理系统,并可集合系统强大的新闻、产品、下载、投票、人才、留言、在线订购、搜索引擎优化、等功能模块,为企业部门提供一个简单、易用、开放、可扩展的企业信息门户平台或电子商务运行平台。开发人员为脆弱页面专门设计了防刷新系统,自动阻止恶意访问和攻击;安全检查应用于每一处代码中,每个提交到系统查询语句中的变量都经过过滤,可自动屏蔽恶意攻击代码,从而全面防

下载
  • 输入预处理别塞进 API handler:把 tokenizer.encode 移到请求前或用 tokenizers 库的 Rust 版本(tokenizers.Tokenizer
  • 禁用 FastAPI 的默认 JSON 响应体校验,加 response_class=Response 并手动 json.dumps(..., separators=(',', ':'))
  • 不要用 time.sleep()logging.info() 在主路径打点,日志写磁盘是同步阻塞操作

显存没爆但 GPU 利用率始终低于 30%

不是模型小,是数据搬运成了瓶颈:CPU 加载 → 预处理 → 拷贝到 GPU → 推理 → 拷回 CPU → 返回 JSON,每步都在等。

立即学习Python免费学习笔记(深入)”;

  • pin_memory=True 创建 DataLoader,配合 non_blocking=True.to('cuda') 时异步传输
  • 把 tokenizer 输出直接转成 torch.tensor(..., device='cuda'),跳过中间 CPU tensor
  • 如果用 Triton 推理服务器,确保 max_batch_sizepreferred_batch_size 匹配真实流量分布,否则小 batch 会等凑够再发
实际压测时发现,transformers 默认的 pad_to_multiple_of=8 在短文本场景反而引入冗余 padding,关掉它比调 batch size 影响更大。还有就是别信文档里“自动优化”的说法——每个 model.forward 调用前,自己 print 出 CUDA event 时间戳,才看得清哪一环真卡。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
C++系统编程内存管理_C++系统编程怎么与Rust竞争内存安全
C++系统编程内存管理_C++系统编程怎么与Rust竞争内存安全

C++系统编程中的内存管理是指 对程序运行时内存的申请、使用和释放进行精细控制的机制,涵盖了栈、堆、静态区等不同区域,开发者需要通过new/delete、智能指针或内存池等方式管理动态内存,以避免内存泄漏、野指针等问题,确保程序高效稳定运行。它核心在于开发者对低层内存有完全控制权,带来灵活性,但也伴随高责任,是C++性能优化的关键。

13

2025.12.22

Rust异步编程与Tokio运行时实战
Rust异步编程与Tokio运行时实战

本专题聚焦 Rust 语言的异步编程模型,深入讲解 async/await 机制与 Tokio 运行时的核心原理。内容包括异步任务调度、Future 执行模型、并发安全、网络 IO 编程以及高并发场景下的性能优化。通过实战示例,帮助开发者使用 Rust 构建高性能、低延迟的后端服务与网络应用。

6

2026.02.11

Python Flask框架
Python Flask框架

本专题专注于 Python 轻量级 Web 框架 Flask 的学习与实战,内容涵盖路由与视图、模板渲染、表单处理、数据库集成、用户认证以及RESTful API 开发。通过博客系统、任务管理工具与微服务接口等项目实战,帮助学员掌握 Flask 在快速构建小型到中型 Web 应用中的核心技能。

97

2025.08.25

Python Flask Web框架与API开发
Python Flask Web框架与API开发

本专题系统介绍 Python Flask Web框架的基础与进阶应用,包括Flask路由、请求与响应、模板渲染、表单处理、安全性加固、数据库集成(SQLAlchemy)、以及使用Flask构建 RESTful API 服务。通过多个实战项目,帮助学习者掌握使用 Flask 开发高效、可扩展的 Web 应用与 API。

75

2025.12.15

json数据格式
json数据格式

JSON是一种轻量级的数据交换格式。本专题为大家带来json数据格式相关文章,帮助大家解决问题。

448

2023.08.07

json是什么
json是什么

JSON是一种轻量级的数据交换格式,具有简洁、易读、跨平台和语言的特点,JSON数据是通过键值对的方式进行组织,其中键是字符串,值可以是字符串、数值、布尔值、数组、对象或者null,在Web开发、数据交换和配置文件等方面得到广泛应用。本专题为大家提供json相关的文章、下载、课程内容,供大家免费下载体验。

544

2023.08.23

jquery怎么操作json
jquery怎么操作json

操作的方法有:1、“$.parseJSON(jsonString)”2、“$.getJSON(url, data, success)”;3、“$.each(obj, callback)”;4、“$.ajax()”。更多jquery怎么操作json的详细内容,可以访问本专题下面的文章。

323

2023.10.13

go语言处理json数据方法
go语言处理json数据方法

本专题整合了go语言中处理json数据方法,阅读专题下面的文章了解更多详细内容。

81

2025.09.10

Golang 生态工具与框架:扩展开发能力
Golang 生态工具与框架:扩展开发能力

《Golang 生态工具与框架》系统梳理 Go 语言在实际工程中的主流工具链与框架选型思路,涵盖 Web 框架、RPC 通信、依赖管理、测试工具、代码生成与项目结构设计等内容。通过真实项目场景解析不同工具的适用边界与组合方式,帮助开发者构建高效、可维护的 Go 工程体系,并提升团队协作与交付效率。

1

2026.02.24

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.7万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号