netease-youdao / BCEmbedding

Netease Youdao's open-source embedding and reranker models for RAG products.
Apache License 2.0
1.37k stars 90 forks source link

笔记本按照demo安装,安装后运行时报错找不到指定的模块。 Error loading "C:\Users\Lenovo\anaconda3\envs\bce\lib\site-packages\torch\lib\fbgemm.dll" or one of its dependencies. #77

Closed chuangzhidan closed 1 month ago

chuangzhidan commented 1 month ago

File "C:\Users\Lenovo\anaconda3\envs\bce\lib\site-packages\BCEmbedding__init.py", line 8, in from .models import * File "C:\Users\Lenovo\anaconda3\envs\bce\lib\site-packages\BCEmbedding\models\init.py", line 8, in from .embedding import EmbeddingModel File "C:\Users\Lenovo\anaconda3\envs\bce\lib\site-packages\BCEmbedding\models\embedding.py", line 9, in import torch File "C:\Users\Lenovo\anaconda3\envs\bce\lib\site-packages\torch\init__.py", line 148, in raise err OSError: [WinError 126] 找不到指定的模块。 Error loading "C:\Users\Lenovo\anaconda3\envs\bce\lib\site-packages\torch\lib\fbgemm.dll" or one of its dependencies.

只支持GPU,cuda吗

shenlei1020 commented 1 month ago

这是pytorch的问题,可以试试安装cpu版的torch,BCEmbedding可以纯cpu推理