1
1
1
DeepSeek
关于vllm引擎参数的选择
2025-04-21 11:10:29
125
vllm serve /data/model/Qwen/QwQ-32B \
 --host 127.0.0.1 \
 --port 2025 \
 --served-model-name QwQ-32B \
 --gpu-memory-utilization 0.9 \
 --max-model-len 30728 \
 --enforce-eager \
 --trust-remote-code \
 --dtype bfloat16 \
 --disable-log-stats \
 --disable-log-requests \
 --api-key gdca  \
--tensor-parallel-size 2 \
 --pipeline-parallel-size 2

目前使用的是欧拉操作系统,vllm 0.6.2。硬件是K100 AI * 3 。
如果是加载的DeepSeek-R1-Distill-Qwen-32B,无论用的  tensor-parallel-size 2 还是  tensor-parallel-size 1 都可以运行,效果不一样。猜测是因为这个模型不会占满单个GPU的64G显存。
但是如果换成QWQ,在用  tensor-parallel-size 1 的时候,会提示HIP out of memory.如果

  tensor-parallel-size 2 则会提示please ensure that world_size (4) is less than than max local gpu count (3)。如何正确的使用这3块DCU正确的加载QWQ(Loading model weights took 61.0392 GB),推测这个模型会需要超过单卡64G空间。

全部评论(5)
只看楼主
H
hepj
1楼
please ensure that world_size (4) is less than than max local gpu count (3)应该是哪里设置有问题设置成了四张卡,但本机只有3张卡。其实32B模型两卡应该就可以推理。可以export HIP_VISIBLE_DEVICES=0,1 然后tensor-parallel-size 2 设置成两张卡进行推理
2025/4/22
R
Ray
回复 @hepj
这样会不会导致一张卡的算力被浪费?或者当有需求加载显存要求超过130G的大模型的时候,需要3个卡的显存共享才能加载,这种情况下还能跨卡加载吗,我查资料都说现在基本上transformer类的模型都支持任意分片,但是我设置3却会报错
2025/4/22
AuroraUSTC
回复 @hepj
如果一定要用上三张卡的话可以用--pipeline-parallel-size 3 + --tensor-parallel-size 1,不过不大推荐这么做,我之前试的时候pipeline parallelism的性能是远不如tensor parallelism的
2025/4/22
R
Ray
回复 @hepj
这我倒是可以理解,我现在主要担心的是显存共享的问题,因为我设置tensor-parallel-size 1会提示显存不足,如果无法跨显卡共享显存加载大模型的话,那相当于限制死了我能加载的大模型的规模
2025/4/22
全部4条回复
没有更多内容了
--
楼主
Level.4
0 帖子
楼主热帖
暂无数据