讲师中心 微信公众号

工具推荐

视频效率加速
Seko
Seko

AI短剧一键生成成片

Laper
Laper

剧本策划到成片协作

立刻MV
立刻MV

低成本歌曲 MV、视频生成

AniShort
AniShort

AI短剧协同制作平台

Python 语义搜索的 Sentence-Transformers 部署

落伟同学_4881

落伟同学_4881

发布时间:2026-02-13 23:01:03

|

449人浏览过

|

来源于php中文网

原创

根本原因是默认float32加载大模型致显存溢出;应改用float16、禁用padding预分配、分批处理;相似度不准因归一化重复或模型不一致;部署需单例加载、启用fast tokenizer;Windows下慎用torch.compile,优先ONNX+CPU推理。

python 语义搜索的 sentence-transformers 部署

模型加载慢、显存爆掉,model.encode() 卡住不动

根本原因不是代码写错了,而是默认用 torch.float32 加载 500MB+ 的模型(比如 sentence-transformers/all-MiniLM-L6-v2),显存吃满后触发 CPU 回退或 OOM。别急着换显卡,先改加载方式。

  • 强制用 torch.float16:加参数 device='cuda',torch_dtype=torch.float16(注意不是 fp16=True,那个只影响推理,不缩模型权重)
  • 禁用 tokenizer 的 padding 预分配:传 tokenizer_kwargs={'padding': False, 'truncation': True},否则 batch 大时内存翻倍
  • 小批量处理:别一次性 encode(list_of_10000_texts),拆成每批 64 或 128 条,用 batch_size=64 参数控制

搜索结果不准,cos_sim() 返回的相似度全在 0.8 以上

不是模型不行,是文本预处理和向量归一化没对齐。Sentence-Transformers 的 encode() 默认已做 L2 归一化,但很多人自己算余弦相似度时又手动归一,导致数值失真。

  • 直接用库里的 util.cos_sim(),别手写 np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b))
  • query 和 doc 必须用同一个模型 encode,别 query 用 all-MiniLM-L6-v2,doc 用 multi-qa-MiniLM-L6-dot-v1 —— 向量空间不兼容,相似度无意义
  • 中文文本务必确认 tokenizer 是否支持:比如 all-MiniLM-L6-v2 是多语言但非中文特化,遇到“微信”“支付宝”这类词嵌入偏弱,可换 paraphrase-multilingual-MiniLM-L12-v2

Docker 部署后 API 响应延迟高,curl 测出来首字节要 2s+

核心瓶颈常在模型首次加载 —— 每个请求都走一遍 SentenceTransformer(...),而不是复用实例。FastAPI/Flask 里别把 model 当局部变量。

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载
  • 全局单例加载:在模块顶层 or app.on_event("startup") 里初始化 model,确保只 load 一次
  • 关闭 tokenizer 的 slow tokenizers:加 use_fast=True 到 model 初始化参数,否则 Hugging Face 的 slow tokenizer 会拖慢首条请求 300ms+
  • 验证是否真缓存:启动后立刻发一条请求,再 time curl http://localhost:8000/search?q=test,第二次响应应 ≤200ms;如果还是慢,检查是否误写了 model = SentenceTransformer(...) 在 route 函数里

Windows 下 torch.compile() 报错或无效

torch.compile() 在 Windows 上目前仅支持 CUDA 12.1+ 且需 nightly 版本 PyTorch,生产环境别硬上。更稳的提速方案是提前编译推理图 + 关闭梯度。

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

  • 部署前加 model.eval() 和 torch.no_grad()(后者必须显式写,不能只靠 eval)
  • 用 torch.jit.script(model) 编译(注意:只支持部分模型结构,all-MiniLM-L6-v2 可行;若报错就跳过)
  • Windows 用户优先用 onnxruntime:导出 ONNX 后用 CPU 推理,比原生 torch 快 1.5–2x,且无平台兼容问题

最易被忽略的是:模型路径含中文或空格时,SentenceTransformer() 在 Windows 下可能静默失败,日志里也不报错 —— 改用绝对路径并确保全是 ASCII 字符。

热门AI工具

更多
Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

讯飞智文

一款面向学习和办公场景的AI文档创作工具,可辅助生成PPT与Word文档,提高资料整理和内容制作效率。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

相关专题

更多
flask框架如何搭建
flask框架如何搭建

搭建步骤:1、安装Python和Pip;2、创建虚拟环境;3、安装Flask;4、创建Flask应用;5、运行应用;6、访问应用。想了解更多flask框架的相关内容,可以阅读本专题下面的文章。

4275

2024.06.27

Python Flask框架
Python Flask框架

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

6624

2025.08.25

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

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

336

2025.12.15

Python FastAPI异步API开发_Python怎么用FastAPI构建异步API
Python FastAPI异步API开发_Python怎么用FastAPI构建异步API

Python FastAPI 异步开发利用 async/await 关键字,通过定义异步视图函数、使用异步数据库库 (如 databases)、异步 HTTP 客户端 (如 httpx),并结合后台任务队列(如 Celery)和异步依赖项,实现高效的 I/O 密集型 API,显著提升吞吐量和响应速度,尤其适用于处理数据库查询、网络请求等耗时操作,无需阻塞主线程。

119

2025.12.22

Python 微服务架构与 FastAPI 框架
Python 微服务架构与 FastAPI 框架

本专题系统讲解 Python 微服务架构设计与 FastAPI 框架应用,涵盖 FastAPI 的快速开发、路由与依赖注入、数据模型验证、API 文档自动生成、OAuth2 与 JWT 身份验证、异步支持、部署与扩展等。通过实际案例,帮助学习者掌握 使用 FastAPI 构建高效、可扩展的微服务应用,提高服务响应速度与系统可维护性。

594

2026.02.06

Python Web框架FastAPI 全栈开发教程合集
Python Web框架FastAPI 全栈开发教程合集

以 FastAPI 为核心,讲解现代 Python Web API 的高效开发方式,涵盖路由定义与路径参数/查询参数/请求体绑定、Pydantic 模型的数据校验与序列化、依赖注入(Depends)系统的分层设计、中间件与 CORS 配置、OAuth2 + JWT 认证流程、后台任务(BackgroundTasks)、WebSocket 实时通信、SQLAlchemy 异步 ORM 集成、自动生成 OpenAPI/Swagger 交互文

736

2026.05.09

Python FastAPI异步微服务与高性能接口设计
Python FastAPI异步微服务与高性能接口设计

本专题聚焦 Python FastAPI 框架在高性能接口与微服务开发中的应用,讲解异步请求处理、依赖注入机制、路由设计、数据库异步操作以及接口性能优化策略。结合实际项目案例,帮助开发者构建高并发、低延迟的现代化后端服务架构。

619

2026.06.16

curl_exec
curl_exec

curl_exec函数是PHP cURL函数列表中的一种,它的功能是执行一个cURL会话。给大家总结了一下php curl_exec函数的一些用法实例,这个函数应该在初始化一个cURL会话并且全部的选项都被设置后被调用。他的返回值成功时返回TRUE, 或者在失败时返回FALSE。

3883

2023.06.14

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

60

2026.11.06

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn