Unanswered
Hi! I'M Running Launch_Multi_Mode With Pytorch-Lightning
I think we need to set more env var if we are running with multiple gpus on 1 node.
Can you try setting:
os.environ["NODE_RANK"] = current_conf["node_rank"] // gpus
os.environ["LOCAL_RANK"] = current_conf["node_rank"] % gpus
os.environ["GLOBAL_RANK"] = current_conf["node_rank"]
51 Views
0
Answers
5 months ago
5 months ago