1
数学库
提问帖
有回复
推理性能测试对比
2026-03-02 15:51:47
183
使用英伟达A40、K100和BW1000单张卡部署Qwen3-0.6B模型,进行benchmark性能测试,设置单卡且1个并发 部署命令: ``` vllm serve /root/Qwen3-0.6B --served-model-name Qwen3-0.6B --trust-remote-code --tensor-parallel-size 1 --dtype float16 --max-model-len 13000 --gpu-memory-utilization 0.9 --enable-prefix-caching ``` 测试命令: ``` python benchmark_serving.py --dataset-name random --tokenizer ./Qwen3-0.6B --model Qwen3-0.6B --random-input-len 256 --random-output-len 512 --num-prompts 256 --ignore-eos --max_concurrency 1 ``` 测试结果如下: | **显卡类型(数量1)** | **prompt数量** | **TTFT(ms)** | **TPOT(ms)** | **latency(ms)** | | -------------------------- | -------------------- | ------------------ | ------------------ | --------------------- | | **K100\_AI(64GB)** | **256** | **34.07** | **20.96** | **20.96** | | **A40(46GB)** | **256** | **29.22** | **6.06** | **6.07** | | **BW 1000(64GB)** | **256** | **13.90** | **6.97** | **6.97** | 这个测试结果有点不尽人意啊,在信息介绍中K100的显存和带宽都要高于A40,但是PD阶段的指标都不理解,虽说VLLM与cuda完美适配,但也不至于差这么多吧,还是我的测试过程有问题,求解答
K100
全部评论(3)
只看楼主
Y
yangzhong
1楼
我使用vllm092镜像环境,对齐您的部署和测试命令,K100AI测出来的各项都要优于A40,您可以使用vllm较新镜像环境复测一下看看
3/5
S
suncd
回复 @yangzhong
我是在不同的机器上测的,超算上的K100的CPU核数只有6个,A40是自己的机器,CPU核数有56个,用profile分析了下,内存拷贝占用时间较长,可能跟这个有关系
3/5
S
suncd
回复 @yangzhong
另外,我在分析profile时,Memcpy DtoH,A40 明确使用了 Pinned -> Device 和 Device -> Pageable,而 K100 中仅显示 Host -> Device 和 Device -> Host,说明可能未启用页锁定内存,内存拷贝效率极低,这个应该怎么设置
3/5
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据