vllm+qwen2部署!
创始人
2024-09-26 17:21:02
0

 准备好qwen2模型:去huggingface镜像、魔搭 都可下载:
HF-Mirror、魔搭社区


创建conda环境:
conda create -n name python==3.10  (python环境一定要3.10 后面有用!


激活环境:
conda activate name


替换镜像源:
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple


安装所需依赖:

pip install modelscope==1.11.0
pip install openai==1.17.1

pip/pip3 install torch torchvision torchaudio pip install tqdm==4.64.1 pip install transformers==4.39.3

安装flash-attn依赖包的时候有坑!
需要先安装nijia这个包:

pip install ninja 
检查ninja是否安装成功:

echo $?

返回0代表安装成功!

此时再次安装flash-attn:

MAX_JOBS=8 pip install flash-attn --no-build-isolation

还是报错,加上代理再次安装!

pip install https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.2/flash_attn-2.5.2+cu122torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
如果超时 可以设置参数 --timeout=250(具体多少根据实际情况定) 
 

参考:安装flash-attention失败的终极解决方案_building wheels for collected packages: flash-attn-CSDN博客

安装成功!



pip install vllm

启动openai风格接口:
python -m vllm.entrypoints.openai.api_server --model /dfs/data/autodl-tmp/qwen/Qwen2-7B-Instruct  --served-model-name Qwen2-7B-Instruct --max-model-len=2048

--dtype=half (我当前显卡为esla V100-PCIE-32GB GPU具有计算能力7.0,不够8.0,所以需要设置半精度,使用float16(half precision)而非Bfloat16进行计算,这样可以降低算力要求)


若想启动多Gpu再设置以下两个参数:
CUDA_VISIBLE_DEVICES=0,1,2,3

并行计算参数:
--tensor-parallel-size=2(张量并行参数设置)
--pipeline-parallel-size=4(管道并行参数设置)

CUDA_VISIBLE_DEVICES=0,1,2,3 python -m vllm.entrypoints.openai.api_server --model /root/autodl-tmp/qwen/Qwen2-7B-Instruct --served-model-name Qwen2-7B-Instruct --max-model-len=2048

成功启动服务!

相关内容

热门资讯

科普“丫丫陕西打锅子辅助”介绍... 科普“丫丫陕西打锅子辅助”介绍开挂作弊辅助软件(最初存在有挂);丫丫陕西打锅子辅助软件透明挂是一个全...
必备科技!长乐茶馆免费辅助器,... 必备科技!长乐茶馆免费辅助器,微信小程序游戏破解微乐游戏,wpk教程(有挂总结);1、必备科技!长乐...
实测揭晓“游戏辅助神器app”... 实测揭晓“游戏辅助神器app”了解开挂作弊辅助助手(原来真的是有挂),游戏辅助神器app是用手机号来...
玩家必用!火神大厅辅助器,钱塘... 您好:钱塘十三水辅助器软件这款游戏可以开挂的,确实是有挂的,很多玩家在这款游戏中打牌都会发现很多用户...
传递经验“宝宝游戏辅助”专业开... 传递经验“宝宝游戏辅助”专业开挂作弊辅助黑科技(一直真的有挂)是一款可以让一直输的玩家,快速成为一个...
玩家必看分享!奇迹手机辅助脚本... 玩家必看分享!奇迹手机辅助脚本可以下载,微信插件激活码,软件教程(详细教程);1、点击下载安装,奇迹...
实测教程“瓜瓜丰城双剑旧版攻略... 实测教程“瓜瓜丰城双剑旧版攻略”讲解开挂作弊辅助黑科技(竟然是真的有挂);瓜瓜丰城双剑旧版攻略最新版...
揭秘一下!牌乐门黑科技,天天填... 您好:天天填大坑辅助器免费下载这款游戏可以开挂的,确实是有挂的,很多玩家在这款游戏中打牌都会发现很多...
必看攻略“小程序广东雀神智能插... 您好:小程序广东雀神智能插件安装这款游戏可以开挂的,确实是有挂的,很多玩家在这款游戏中打牌都会发现很...
查到实测辅助!花花生活圈可以挂... 查到实测辅助!花花生活圈可以挂辅助,游戏老友地方有脚本,解密教程(有挂神器);1、全新机制【花花生活...