1
6
1
DeepSeek
新人贴
使用k100部署bge-m3和reranker模型
2026-04-07 17:06:06
349
1、使用docker命令拉景象: 执行命令:docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:vllm0.7.2-ubuntu22.04-dtk25.04-rc7-das1.5-py3.10-20250327 执行:docker images 找到对应下载的镜像ID vllm0.7.2-ubuntu22.04-dtk25.04-rc7-das1.5-py3.10-20250327 42295ada4800 我下载的images id为42295ada4800 2、部署bge-m3向量分词模型 1)执行 docker run -it --shm-size 16g --network=host --name bge-m3 --privileged --device=/dev/kfd --device=/dev/dri --device=/dev/mkfd --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root -v /opt/hyhal/:/opt/hyhal/:ro -v /data1/:/workspace/ 42295ada4800 bash 2)进入docker对应的镜像bge-m3中 通过modelscope下载向量分词模型 https://modelscope.cn/models/BAAI/bge-m3/files 3)安装modelscope 执行pip install modelscope 4)通过modelscope下载bge-m3 modelscope download --model BAAI/bge-m3 5)将下载好的模型拷贝到/workspace/models/bge-m3/ 目录下 6)执行启动模型命令: HIP_VISIBLE_DEVICES=0 python3 -m vllm.entrypoints.openai.api_server --model /workspace/models/bge-m3/ --dtype float16 --max-model-len 2048 --port 8003 --served-model-name bgem3 --task embedding --tensor-parallel-size 1 > vllm_bgem3.log 2>&1 & 7)查看是否安装成功,查看运行日志: tail -200f vllm_bgem3.log 3、部署bge-reranker 向量分词模型 1)执行 docker run -it --shm-size 16g --network=host --name bge-reranker --privileged --device=/dev/kfd --device=/dev/dri --device=/dev/mkfd --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root -v /opt/hyhal/:/opt/hyhal/:ro -v /data1/:/workspace/ 42295ada4800 bash 2)进入docker对应的镜像bge-reranker 中 通过modelscope下载向量重排模型 https://modelscope.cn/models/BAAI/bge-reranker-v2-m3 3)安装modelscope 执行pip install modelscope 4)通过modelscope下载bge-reranker-v2-m3 modelscope download --model BAAI/bge-reranker-v2-m3 5)将下载好的模型拷贝到/workspace/models/bge-reranker-v2-m3/ 目录下 6)执行启动模型命令: HIP_VISIBLE_DEVICES=0 python3 -m vllm.entrypoints.openai.api_server --model /workspace/models/bge-reranker-v2-m3/ --dtype float16 --max-model-len 2048 --port 8002 --served-model-name bge-reranker --task embedding --tensor-parallel-size 1 > vllm_bge-reranker.log 2>&1 & 7)查看是否安装成功,查看运行日志: tail -200f vllm_bge-reranker.log ![attachmentId-image.png](/api-forum/file/RedirectImage?path=20260407/1775552759610_image.png)
全部评论(2)
只看楼主
L
lim
1楼
:guzhang:emoji
4/7
T
tony250
回复 @lim
试了--task embedding这个参数不行啊
4/21
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据