diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index f153d6a9..16338e17 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -380,6 +380,9 @@ class Trainer(TrainerIO): # otherwise we launch the required number of processes try: nb_slurm_tasks = int(os.environ['SLURM_NTASKS']) + print('x'*100) + print(f'NB_TASKS: {nb_slurm_tasks}') + print('x'*100) nb_requested_gpus = len(self.data_parallel_device_ids) is_slurm_managing_tasks = nb_slurm_tasks == nb_requested_gpus except Exception as e: