wehub-resource-sync
17 lines
394 B
Python
17 lines
394 B
Python
# Copyright (c) Microsoft Corporation.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# DeepSpeed Team
|
|
|
|
import os
|
|
from unit.common import get_master_port
|
|
|
|
|
|
def setup_serial_env():
|
|
# Setup for a serial run
|
|
os.environ['MASTER_ADDR'] = '127.0.0.1'
|
|
os.environ['MASTER_PORT'] = get_master_port()
|
|
os.environ['LOCAL_RANK'] = '0'
|
|
os.environ['RANK'] = '0'
|
|
os.environ['WORLD_SIZE'] = '1'
|