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数据抓包,wp... 《推荐攻略!wpk数据抓包,wpk德州扑克靠不靠谱,必备教程(真是有挂)-哔哩哔哩》 wpk德州扑克...
七分钟俱乐部!wepoke有a... 七分钟俱乐部!wepoke有app软件吗(透明挂黑科技)外挂透明挂辅助神器(2024已更新)(哔哩哔...
今日焦点!(Wepoke工具)... 今日焦点!(Wepoke工具)软件透明挂辅助插件!(鱼扑克辅助)可靠教程(2024已更新)(哔哩哔哩...
六分钟科普!福建大菠萝万能辅助... 六分钟科普!福建大菠萝万能辅助器,aapoker透视脚本下载(透视)安装教程(新版有挂)1、福建大菠...
一分钟了解!wepoke用模拟... 一分钟了解!wepoke用模拟器有可以用,微扑克辅助软件下载,透明教程(有挂头条)-哔哩哔哩;wep...
两分钟科普!wpk辅助透视技能... 两分钟科普!wpk辅助透视技能教程(透明挂黑科技)外挂透明挂辅助机器人(2023已更新)(哔哩哔哩)...
一分钟带你了解!(wEpoKe... 一分钟带你了解!(wEpoKe)外挂辅助软件!(德州版Wepoke)安装教程(2023已更新)(哔哩...
第9分钟神器!aapoker俱... 第9分钟神器!aapoker俱乐部靠谱吗,wepokerplus透视挂(透视)靠谱教程(确实有挂)1...
实测发现!微扑克统计辅助器,a... 实测发现!微扑克统计辅助器,aapoker有假,wpk教程(有挂方法)-哔哩哔哩;微扑克统计辅助器最...
三分钟领会!德扑ai智能机器人... 三分钟领会!德扑ai智能机器人平台安装(透明挂黑科技)外挂透明挂辅助黑科技(2025已更新)(哔哩哔...