# 设置 GPU 的总数量
GPUS=1
# 设置每个节点分配的 GPU 数量
GPUS_PER_NODE=1
# 计算节点数量：总 GPU 数除以每个节点的 GPU 数
NNODES=$(expr $GPUS / $GPUS_PER_NODE)
# 设置当前节点的编号，如果环境变量 RANK 没有设置，则默认赋值为 0
RANK=${RANK:-0}
# 如果环境变量 MASTER_ADDR 没有设置，则默认使用开发机地址 "127.0.0.1"
MASTER_ADDR=${MASTER_ADDR:-"127.0.0.1"}
# 如果环境变量 MASTER_PORT 没有设置，则默认赋值为 29850
MASTER_PORT=${MASTER_PORT:-29850}