我们使用了由@pakerface的@BMY顺其自然维护的Sentence Transformers库,以在多个GPU上分布式处理嵌入过程。你可以在这里找到一个关于如何使用多进程进行嵌入的示例:https://github.com/UKPLab/sentence-transformers/blob/66e0ee30843dd411c64f37f65447bb38c7bf857a/examples/applications/computing-embeddings/computing_embeddings_multi_gpu.py