安装
我们推荐使用 Python 3.10+、PyTorch 1.11.0+ 和 transformers v4.41.0+。Sentence Transformers 有 5 个额外的安装选项
默认： 允许加载、保存和推理（即获取嵌入）模型。
ONNX： 允许使用 ONNX 后端加载、保存、推理、优化和量化模型。
OpenVINO： 允许使用 OpenVINO 后端加载、保存和推理模型。
默认和训练：与 默认 相同，但增加了训练功能。
开发：包括上述所有功能，以及开发 Sentence Transformers 所需的一些依赖项，请参阅 可编辑安装。
请注意，您可以混合搭配各种额外选项，例如 pip install -U "sentence-transformers[train,onnx-gpu]"。
使用 pip 安装
pip install -U sentence-transformers
适用于 GPU 和 CPU
pip install -U "sentence-transformers[onnx-gpu]"
仅适用于 CPU
pip install -U "sentence-transformers[onnx]"
pip install -U "sentence-transformers[openvino]"
pip install -U "sentence-transformers[train]"
要使用 Weights and Biases 跟踪您的训练日志，您还应该安装 wandb （推荐）
pip install wandb
为了在训练时跟踪您的碳排放，并将此信息自动包含在您的模型卡中，您还应该安装 codecarbon （推荐）
pip install codecarbon
pip install -U "sentence-transformers[dev]"
使用 Conda 安装
conda install -c conda-forge sentence-transformers
适用于 GPU 和 CPU
pip install -U "sentence-transformers[onnx-gpu]"
仅适用于 CPU
pip install -U "sentence-transformers[onnx]"
pip install -U "sentence-transformers[openvino]"
conda install -c conda-forge sentence-transformers accelerate datasets
要使用 Weights and Biases 跟踪您的训练日志，您还应该安装 wandb （推荐）
pip install wandb
为了在训练时跟踪您的碳排放，并将此信息自动包含在您的模型卡中，您还应该安装 codecarbon （推荐）
pip install codecarbon
conda install -c conda-forge sentence-transformers accelerate datasets pre-commit pytest ruff
从源代码安装
您可以直接从源代码安装 sentence-transformers，以利用最新的 main 分支，而不是最新的稳定版本
pip install git+https://github.com/huggingface/sentence-transformers.git
适用于 GPU 和 CPU
pip install -U "sentence-transformers[onnx-gpu] @ git+https://github.com/huggingface/sentence-transformers.git"
仅适用于 CPU
pip install -U "sentence-transformers[onnx] @ git+https://github.com/huggingface/sentence-transformers.git"
pip install -U "sentence-transformers[openvino] @ git+https://github.com/huggingface/sentence-transformers.git"
pip install -U "sentence-transformers[train] @ git+https://github.com/huggingface/sentence-transformers.git"
要使用 Weights and Biases 跟踪您的训练日志，您还应该安装 wandb （推荐）
pip install wandb
为了在训练时跟踪您的碳排放，并将此信息自动包含在您的模型卡中，您还应该安装 codecarbon （推荐）
pip install codecarbon
pip install -U "sentence-transformers[dev] @ git+https://github.com/huggingface/sentence-transformers.git"
可编辑安装
如果您想对 sentence-transformers 进行更改，您将需要一个可编辑安装。克隆存储库并使用以下命令安装它
git clone https://github.com/huggingface/sentence-transformers
cd sentence-transformers
pip install -e ".[train,dev]"
这些命令将链接新的 sentence-transformers 文件夹和您的 Python 库路径，以便在导入 sentence-transformers 时使用此文件夹。
安装支持 CUDA 的 PyTorch
要使用 GPU/CUDA，您必须安装支持 CUDA 的 PyTorch。请按照 PyTorch - 入门 中的安装步骤进行操作。