显卡是GeForce RTX 3090。CUDA 11.2,python 版本是3.7.1。 其实,说来也怪,用了好多torch,有1.7.1, 1.7.0 ,1.8.1都可以在其他3090(OS Ubantu 20.04)上跑,当然,不是这台服务器(os UbanTu 20.04)。不知道有没有大佬指点,涉及到这方面,就是卡死我。 先说一下,配置torch吧,其实不难。去这里进去官...
此外,还需要检查你的代码是否有其他错误,比如拼写错误、导入语句错误等。 如果以上步骤都无法解决问题,建议查看flagembedding模块的官方文档或联系库的维护者以获取帮助。同时,确保你的Python和pip都是最新版本,有时候旧版本的Python或pip可能会导致一些兼容性问题。
在尝试使用example中的MS MARCO评测脚本时遇到了如下问题: File "/flagemb/FlagEmbedding/baai_general_embedding/finetune/eval_msmarco.py", line 119, in index faiss_index.add(corpus_embeddings) line 230, in replacement_add self.add_c(n, swig_ptr(x)) File "/.conda/envs/emb/lib/python3.10/...
@545999961作者,你好。我这边按照你的指示加了gradient_checkpoint和deepspeed,我的环境是: torch 2.4.0+cu118;torchaudio 2.4.0+cu118;torchvision 0.19.0+cu118;deepspeed 0.15.3;python 3.9 目前的报错是: 麻烦作者帮忙看看呢,谢谢~ 看起来像是deepspeed的问题,可以尝试更新deepspeed或更换使用的deepspeed配置文件 ...
不关注关注所有动态仅关注版本发行动态关注但不提醒动态 1Star0Fork0 渣渣cong/FlagEmbedding 代码Issues0Pull Requests0Wiki统计流水线 服务 加入Gitee 与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :) 免费加入 已有帐号?立即登录
本机环境:python=3.8,pytorch=1.11.0,nvidia-smi=450.119.04,cuda version=11.0,nvcc -V=11.3.109 执行pip install FlagEmbedding成功 python控制台执行 from FlagEmbedding import FlagReranker reranker = FlagReranker('/root/workspace/bge-reranker-v2-m3', use_f