可参考的环境变量设置: 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

暂无数据