Skip to content

Commit ae2cd9b

Browse files
committed
Update accelerator_connector.py
1 parent 19e67d1 commit ae2cd9b

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

pytorch_lightning/trainer/connectors/accelerator_connector.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -112,12 +112,6 @@ def __init__(
112112
self._training_type_plugin: Optional[TrainingTypePlugin] = None
113113
self._cluster_environment: Optional[ClusterEnvironment] = None
114114

115-
# init the default rank if exists
116-
# we need to call this here or NVIDIA flags and other messaging in init will show on all ranks
117-
# this way we only show it on rank 0
118-
if "LOCAL_RANK" in os.environ:
119-
rank_zero_only.rank = int(os.environ["LOCAL_RANK"])
120-
121115
# for gpus allow int, string and gpu list
122116
if auto_select_gpus and isinstance(gpus, int):
123117
self.gpus = pick_multiple_gpus(gpus)

0 commit comments

Comments
 (0)