Sentence Transformers 安装指南:pip/uv/Conda 全流程与各功能扩展包(extras)详解

发布时间:2026/9/20 22:31:53

Sentence Transformers 安装指南:pip/uv/Conda 全流程与各功能扩展包(extras)详解 人工智能NLPEmbedding微调【免费下载链接】sentence-transformersState-of-the-Art Embeddings, Retrieval, and Reranking项目地址https://gitcode.com/gh_mirrors/se/sentence-transformers点击查看免费下载本文是 Sentence Transformers当前仓库版本6.2.0.dev0的完整安装实战指南覆盖uv、pip、Conda三种主流包管理方式逐一讲解 Default、Image、Audio、Video、Training、ONNX、OpenVINO、Development 等全部扩展包extras的适用场景与依赖组成并介绍从源码安装、可编辑安装开发模式以及 PyTorch CUDA 环境的配置方法。读完本文你可以根据只做推理、要训练微调、要处理多模态输入、要用 ONNX/OpenVINO 后端加速等不同需求准确选择并完成最小化、可复现的安装。安装前置要求官方推荐的运行环境基线为Python 3.10PyTorch 2.2需按是否使用 GPU 选择对应版本详见下文 安装 PyTorch 并启用 CUDA 一节transformers v5.0这一基线在仓库的 pyproject.toml 中得到了印证项目声明requires-python 3.10核心依赖包含transformers5.0.0,6.0.0、tokenizers0.19、huggingface-hub1.3.0,3.0.0、torch2.2、numpy1.24.0、scikit-learn1.1.0、scipy1.0.0、typing_extensions4.10.0、tqdm4.0.0。也就是说即使是最小安装默认包pip也会自动解析并装入上述依赖无需手动逐个安装。安装选项extras总览Sentence Transformers 的安装包按功能拆分为若干可选扩展extras其依赖定义见 pyproject.toml 的[project.optional-dependencies]段。你可以根据用途按需取用并且各 extras 之间可以自由组合例如pip install -U sentence-transformers[train,image,video,onnx-gpu]。选项一句话说明实际引入的依赖来自 pyproject.tomlDefault默认文本模型的加载、保存与推理获取 embedding仅核心依赖无额外项Image支持处理图像的模型如 CLIP、基于 VLM 的模型transformers[vision]Audio支持处理音频输入的模型transformers[audio]Video支持处理视频输入的模型transformers[video]Training训练与微调模型所需依赖datasets2.16.0、accelerate1.3.0ONNX使用 ONNX 后端加载/保存/推理/优化/量化模型optimum-onnx[onnxruntime]CPU 版ONNX-GPUONNX 后端的 GPU 加速版本optimum-onnx[onnxruntime-gpu]GPU 版OpenVINO使用 OpenVINO 后端加载/保存/推理模型optimum-intel[openvino]2.0.0Development以上全部能力外加开发 Sentence Transformers 本身所需的依赖见下文 Development 开发环境 一节两个值得注意的细节ONNX 分为 CPU 与 GPU 两种安装方式onnx对应optimum-onnx[onnxruntime]仅含 CPU 版 ONNX Runtimeonnx-gpu对应optimum-onnx[onnxruntime-gpu]额外包含 CUDA 版运行库。两者不要混装。torchcodec 需要单独安装若要以torchcodec.AudioDecoder/torchcodec.VideoDecoder实例作为模型输入必须先单独执行pip install torchcodec。在源码中AudioDecoder与VideoDecoder都是可选导入的见 sentence_transformers/base/modality.py 与 sentence_transformers/base/modality_types.pysentence_transformers/util/environment.py 还专门提示缺少 torchcodec 时 transformers 会回退到 torchvision 已移除的read_video接口因此处理视频输入时务必安装 torchcodec。使用 uv 安装uv 是高性能的 Python 包管理器安装命令以uv pip install -U开头-U表示升级到最新版场景命令默认uv pip install -U sentence-transformers图像uv pip install -U sentence-transformers[image]音频uv pip install -U sentence-transformers[audio]视频uv pip install -U sentence-transformers[video]训练uv pip install -U sentence-transformers[train]ONNXGPU 与 CPUuv pip install -U sentence-transformers[onnx-gpu]ONNX仅 CPUuv pip install -U sentence-transformers[onnx]OpenVINOuv pip install -U sentence-transformers[openvino]开发uv pip install -U sentence-transformers[dev]训练日志与碳排放追踪可选推荐使用train扩展进行训练时官方还推荐安装两个辅助库# 用 Weights Biases 或 Trackio 追踪训练日志推荐二选一 uv pip install trackio # 追踪训练过程的碳排放并自动写入模型卡片 uv pip install codecarbon注意安装之后训练时还必须在 Training Arguments 的report_to参数中加入对应模块名例如wandb、trackio、codecarbon否则这些工具不会被真正启用。使用 pip 安装pip是最通用的安装方式命令与 uv 一一对应场景命令默认pip install -U sentence-transformers图像pip install -U sentence-transformers[image]音频pip install -U sentence-transformers[audio]视频pip install -U sentence-transformers[video]训练pip install -U sentence-transformers[train]ONNXGPU 与 CPUpip install -U sentence-transformers[onnx-gpu]ONNX仅 CPUpip install -U sentence-transformers[onnx]OpenVINOpip install -U sentence-transformers[openvino]开发pip install -U sentence-transformers[dev]训练场景下同样建议追加安装pip install trackio或wandb与pip install codecarbon并记得在 Training Arguments 的report_to中声明相应模块名。扩展包组合示例因为 extras 是可叠加的一份安装即可覆盖多种需求。例如同时需要训练、图像、视频与 ONNX GPU 推理pip install -U sentence-transformers[train,image,video,onnx-gpu]使用 Conda 安装基础包已发布在conda-forge频道上。需要说明的是extras如[image]、[train]是 pip 的概念Conda 并不支持因此使用 Conda 安装基础包后扩展能力仍需通过 pip 补齐。场景命令默认conda install -c conda-forge sentence-transformers图像pip install -U sentence-transformers[image]音频pip install -U sentence-transformers[audio]视频pip install -U sentence-transformers[video]训练conda install -c conda-forge sentence-transformers accelerate datasetsONNXGPU 与 CPUpip install -U sentence-transformers[onnx-gpu]ONNX仅 CPUpip install -U sentence-transformers[onnx]OpenVINOpip install -U sentence-transformers[openvino]开发conda install -c conda-forge sentence-transformers accelerate datasets pre-commit pytest ruff训练场景下同样建议pip install trackio或wandb与pip install codecarbon并在 Training Arguments 的report_to中声明对应模块名。可以看出Conda 路线下train与dev两个场景会直接通过 conda-forge 装好accelerate、datasets等训练/测试相关库而其余多模态与后端相关扩展一律走 pip。从源码安装Source Install如果你想尝鲜main分支上的最新特性而不是等待 PyPI 上的稳定版发布可以直接从源码安装场景命令默认pip install githttps://github.com/huggingface/sentence-transformers.git图像pip install -U sentence-transformers[image] githttps://github.com/huggingface/sentence-transformers.git音频pip install -U sentence-transformers[audio] githttps://github.com/huggingface/sentence-transformers.git视频pip install -U sentence-transformers[video] githttps://github.com/huggingface/sentence-transformers.git训练pip install -U sentence-transformers[train] githttps://github.com/huggingface/sentence-transformers.gitONNXGPU 与 CPUpip install -U sentence-transformers[onnx-gpu] githttps://github.com/huggingface/sentence-transformers.gitONNX仅 CPUpip install -U sentence-transformers[onnx] githttps://github.com/huggingface/sentence-transformers.gitOpenVINOpip install -U sentence-transformers[openvino] githttps://github.com/huggingface/sentence-transformers.git开发pip install -U sentence-transformers[dev] githttps://github.com/huggingface/sentence-transformers.git训练场景下的trackio或wandb与codecarbon附加安装、以及report_to配置要求与前面 pip 章节完全一致。可编辑安装Editable Install开发模式如果你计划修改 Sentence Transformers 的源码并即时验证效果就需要使用可编辑安装。先克隆仓库再以开发依赖安装git clone https://github.com/huggingface/sentence-transformers cd sentence-transformers pip install -e .[train,dev]执行后sentence-transformers目录会与你的 Python 库路径建立链接当你import sentence_transformers时使用的就是本地克隆出来的这份代码你对源码所做的改动无需重装即可生效。这是贡献代码、调试底层逻辑例如 sentence_transformers/backend/load.py 中的后端加载逻辑时的推荐姿势。关于开发依赖devextra从 pyproject.toml 可以看到它实际聚合了训练相关datasets2.16.0、accelerate1.3.0测试相关pytest、pytest-cov、pytest-env、pytest-subtests、pytest-xdist代码质量pre-commit额外能力fastcluster1.2.0、peft、transformers[vision,audio,video]、faiss-cpu、usearch因此[train,dev]的组合足以覆盖训练 测试 多模态 向量检索的完整开发闭环。验证安装是否成功安装完成后可执行如下命令快速验证python -c import sentence_transformers; print(sentence_transformers.__version__)若输出类似6.2.0.dev0的版本号或对应发行版本号说明安装成功。安装 PyTorch 并启用 CUDApip install sentence-transformers默认拉取的是 CPU 版 PyTorch。如果要用 GPU/CUDA 加速必须单独安装带 CUDA 支持的 PyTorch前往 PyTorch 官方 Get Started 页面根据你的 CUDA 版本选择对应的安装命令例如pip3 install torch --index-url https://download.pytorch.org/whl/cu121在安装 PyTorch 之后、安装或升级sentence-transformers或确保pip不会用 CPU 版覆盖掉已装的 CUDA 版 PyTorch安装完成后运行python -c import torch; print(torch.cuda.is_available())检查输出True即代表 CUDA 可用。另外若使用 ONNX GPU 后端请选用sentence-transformers[onnx-gpu]optimum-onnx[onnxruntime-gpu]而不是onnxCPU 版。后端ONNX / OpenVINO与安装包的对应关系如果你打算使用 ONNX 或 OpenVINO 后端加载模型理解源码中的依赖检查逻辑可以帮你避免踩坑ONNX 后端sentence_transformers/backend/load.py 中的load_onnx_model在运行时导入onnxruntime与optimum.onnxruntime若未安装会抛出异常并明确提示安装sentence-transformers[onnx]或sentence-transformers[onnx-gpu]。此外该函数默认按优先级自动选择可用的 provider如 TensorRT CUDA CPU也可通过provider参数手动指定。OpenVINO 后端sentence_transformers/backend/load.py 中的load_openvino_model需要optimum.intel.openvino未安装时同样会提示pip install sentence-transformers[openvino]其加载目标是openvino*.xml权重文件。OpenVINO 静态量化sentence_transformers/backend/quantize.py 中的export_static_quantized_openvino_model额外要求datasets、optimum-intel与openvino缺失时会提示pip install datasets sentence-transformers[openvino]。因此计划使用某一后端前请先按上文表格安装对应的 extras避免运行到一半才报ModuleNotFoundError。安装决策速查你的需求推荐安装命令仅做文本 embedding 推理pip install -U sentence-transformers推理 图像/音频/视频多模态模型pip install -U sentence-transformers[image,audio,video]按需取舍训练 / 微调pip install -U sentence-transformers[train]并视需要追加pip install trackio codecarbon推理 ONNX 加速CPUpip install -U sentence-transformers[onnx]推理 ONNX 加速GPUpip install -U sentence-transformers[onnx-gpu]推理 OpenVINO 加速pip install -U sentence-transformers[openvino]参与开源贡献 / 修改源码git clone后执行pip install -e .[train,dev]GPU 训练或推理先按官方指引安装 CUDA 版 PyTorch再选择上述命令安装完成并验证版本号输出正常后即可进入下一步参考 docs/quickstart.rst 快速上手文本 embedding或阅读 examples/sentence_transformer 下的各类应用示例开始你的 embedding、检索与重排之旅。赞分享人工智能NLPEmbedding微调【免费下载链接】sentence-transformersState-of-the-Art Embeddings, Retrieval, and Reranking项目地址https://gitcode.com/gh_mirrors/se/sentence-transformers点击查看免费下载相关推荐smolagents 安装完全指南pip/uv 安装、可选扩展extras与安装验证smolagents 安装完全指南pip/uv 安装、可选扩展extras与安装验证 本指南基于当前仓库smolagentsHugging Face人工智能AI AgentAgent 框架工具调用代码智能体MCP ClientsAgent 沙箱marimo 安装指南pip、uv、conda 三方式安装与推荐依赖配置详解marimo 安装指南pip、uv、conda 三方式安装与推荐依赖配置详解 marimo 是一个用纯 Python 存储的响应式笔记本reactive n数据科学前端后端AI 应用AutoGluon 安装完全指南pip、uv、Conda、源码四种方式与 CPU/GPU 配置详解AutoGluon 安装完全指南pip、uv、Conda、源码四种方式与 CPU/GPU 配置详解 本文是 AutoGluon 开源自动机器学习框架的官方安装人工智能机器学习AutoML深度学习数据科学创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
延伸阅读

更多相关文章

2026/9/20 22:31:53

自动化专业面试高频考点解析:从控制理论到项目实战经验

简介:东南大学自动化专业复试面试常见问题总结文档,面向考研复试、保研面试以及自动化基础复习人群,集中梳理了信号处理、嵌入式、控制理论、模拟电路、数字电路、通信接口、高数等多个核心方向的典型考点。文档共1个doc文件,压缩…

2026/9/20 22:31:53

gbrain 校准回路(Calibration Loop)开发规范与实践指南

gbrain 校准回路(Calibration Loop)开发规范与实践指南 【免费下载链接】gbrain Garrys Opinionated OpenClaw/Hermes Agent Brain 项目地址: https://gitcode.com/gh_mirrors/gb/gbrain gbrain 是一个将「预测性 Take(观点/判断&…

2026/9/21 0:27:24

Python与Simulink实战:PID控制器从调参到部署

1. 从一个真实翻车现场说起:为什么PID参数总调不好刚入行做控制那会儿,我接手过一个温度控制箱的项目。硬件搭好了,传感器校准了,加热丝也接上了,结果一上电就傻眼——温度要么在目标值附近来回振荡十几度,…

2026/9/21 0:27:24

通达信超级短线指标源码拆解:80%成功率真相与实战调参

简介:这是一份通达信指标公式源码详解PDF,面向短线交易者和技术分析爱好者,聚焦如何借助移动平均线与MACD多条件过滤来捕捉80%以上成功概率的短线买卖信号。包体为单个PDF文件,大小174KB,内容集中不冗长,适…

2026/9/21 0:27:24

基于MATLAB的6节点天然气管网潮流计算程序实现与解析

简介:面向油气储运与能源动力专业的学生,这份基于MATLAB的6节点天然气潮流计算程序,可用于理解天然气输送管网中压力、流量与储存量的非线性求解过程。程序围绕状态方程、节点能量守恒、管道摩擦阻力与压降计算展开,通过迭代算法&…

2026/9/21 0:27:24

AI工具提升学术生产力:9款利器助力论文写作

1. 论文写作新范式:AI工具如何重塑学术生产力去年指导研究生论文时,我发现一个有趣现象:那些善用AI工具的学生,往往能在保证学术严谨性的前提下,将文献综述效率提升3倍以上。这促使我系统测试了市面上137款学术类AI工具…

2026/9/20 0:04:49

GAMP 5 基于风险的计算机化系统验证:软件分类与审计追踪实践

简介:《A Risk-Based Approach to Compliant GxP Computerized Systems》即业内熟知的GAMP 5指南,面向制药企业质量与IT合规人员、验证工程师及计算机化系统管理者,用于解决GxP法规环境下系统合规性难以科学落地的问题。文档以风险管理为主线…

2026/9/20 0:04:49

安全托管MSSP实战:从静态防御到人机协同的攻防运营与应急响应

简介:这份PPT围绕互联网业务安全托管服务展开,面向企业安全负责人、IT运维人员及关注MSSP/MSS选型的读者,重点回应传统安全过度依赖人工、碎片化静态防御难以对抗产业化攻击等痛点。资源共1个pptx文件,包体约30.63MB,以…

2026/9/21 0:02:23

OpenResearch:构建可复现的开放式研究工作流

第一次看到“OpenResearch”这个名字,我脑子里冒出的不是某个具体软件,而更像一种研究方式的宣言:开放、可复现、可验证。这三件事放在一起,其实比大多数人想象中难得多。过去几年我一直在折腾自己的研究工作流,从纯纸…

2026/9/20 4:54:47

USB Type-C PCB布局分区设计:电源、高速信号与PD协议全攻略

做硬件这行,Type-C接口算是典型的“看着简单,做起来全坑”的东西。光引脚就24个,高低速信号、电源、控制线全部塞在一个小小的连接器里,如果PCB布局不做规划,打样回来基本就是“插上没反应”、“高速掉线”、“静电一打…

2026/9/20 5:01:23

系统编程学习原型如何补齐稳定性边界

系统编程学习原型如何补齐稳定性边界预算有限时&#xff0c;我先优化明显多余的复制&#xff0c;而不是猜测性地换容器。用借用传递只读数据通常就能减少分配&#xff1a; fn parse(line: &str) -> Result<Item, Error> { /* ... */ }用基准确认热点确实在分配&am…

2026/9/20 5:09:33

雨花区哪家财务公司代理记账比较好?

在雨花区&#xff0c;企业处理财税事务常常面临诸多挑战&#xff0c;选择一家靠谱的财务公司至关重要。湖南巨勤财务管理咨询有限公司就是本地正规实体财税服务机构&#xff0c;深耕本地工商财税行业多年&#xff0c;熟悉当地工商局、税务局最新政策与申报流程。主营公司注册、…

还想了解更多?直接咨询顾问

免费诊断 + 免费方案 + 透明报价。

全国咨询热线400-8866-253
免费获取方案
咨询二维码