如何开启GPU - Githubissues

shibing624 / text2vec

text2vec, text to vector. 文本向量表征工具，把文本转化为向量矩阵，实现了Word2Vec、RankBM25、Sentence-BERT、CoSENT等文本表征、文本相似度计算模型，开箱即用。

https://pypi.org/project/text2vec/

Apache License 2.0

4.39k stars 392 forks source link

Closed 68cptn closed 1 year ago

68cptn commented 1 year ago

Please provide a clear and concise description of what the question is.

shibing624 commented 1 year ago

训练阶段默认用gpu训练，预测阶段encode 方法有device参数，传入cuda就是gpu