diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 4a357662..e278b1bd 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -165,9 +165,6 @@ if __name__ == '__main__': parser = TRAINING_MODEL.add_model_specific_args(parent_parser, root_dir) hyperparams = parser.parse_args() - # format GPU layout - os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" - # --------------------- # RUN TRAINING # ---------------------