4
框架组件
请问有使用vllm+IB网络在两台机器上部署的案例吗
2025-11-14 14:55:12
174
请问有通过vllm+IB网络部署成功的案例吗,我使用docker镜像 image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10 在两台机器上部署,docker镜像中可以检测到IB设备,但vllm部署时显示无法检测到IB设备。以下是运行指令和ib信息: ![attachmentId-image.png](/api-forum/file/Redirect?path=forum/humingtao/1763102950025/image.png) ![attachmentId-image.png](/api-forum/file/Redirect?path=forum/humingtao/1763102919367/image.png) 运行时NCCL log显示无法检测到IB设备: ![attachmentId-image.png](/api-forum/file/Redirect?path=forum/humingtao/1763103281182/image.png)
开源社区
框架组件
全部评论(2)
只看楼主
W
wangs
1楼
可参考的环境变量设置: export NCCL_MAX_NCHANNELS=16 export NCCL_NET_GDR_READ=1 export HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 export NCCL_IB_HCA=mlx5_2:1,mlx5_3:1,mlx5_4:1,mlx5_5:1,mlx5_6:1,mlx5_7:1,mlx5_8:1,mlx5_9:1 export NCCL_TOPO_FILE="/public/home/gaoqiong/yangql/topo-input.xml" export VLLM_TORCH_PROFILER_DIR=/workspace/int8prof/ export COMPASS_DATA_CACHE=/workspace export VLLM_ENABLE_TBO=0 export VLLM_ZERO_OVERHEAD=0 设置上述环境变量后, 启动ray:head节点执行: ray start --head --node-ip-address=12.12.12.49 --port=6388 --num-gpus=8 --num-cpus=32 worker节点执行: ray start --address='12.12.12.49:6388' --node-ip-address=12.12.12.50 --num-gpus=8 --num-cpus=32
2025/11/21
网
网友lw8uNw
回复 @wangs
请问你这边使用mlx的IB网卡是可以部署的吗,如果是使用shca的IB网卡应该怎么设置呢,我这边没有mlx的网卡,使用shca的网卡NCCL一直检测不到IB设备,是不适配吗?还有就是想问下使用mlx的IB网卡部署的模型推理速度能达到多少tokens/s
2025/11/24
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据