scripts/train_multi_gpus_with_ddp.py

import os
import sys
# sys.path.append(sys.path[0]+'/../')

os.system(
    "CUDA_VISIBLE_DEVICES=1,2,3,4,5,6,7 "
    "torchrun "
    "--nnodes=1 "
    "--nproc_per_node=6 "
    "train_inr_funsr_ddp.py"
)