Installation

We recommend Python 3.6 or higher, PyTorch 1.2.0 or higher and transformers v3.1.0 or higher. The code does not work with Python 2.7. Note, for some features and pre-trained models, you need PyTorch 1.6.0 or higher.

Install SentenceTransformers

Install with pip

Install the sentence-transformers with pip:

pip install -U sentence-transformers

Install from sources

Alternatively, you can also clone the latest version from the repository and install it directly from the source code:

pip install -e .

Install PyTorch with CUDA-Support

If you want to use a GPU / CUDA, you must install PyTorch with the matching CUDA Version. Follow PyTorch - Get Started for further details how to install PyTorch.