diff --git a/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py b/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py
index 8bf1c335..1f2e1aad 100644
--- a/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py
+++ b/examples/new_project_templates/multi_node_examples/multi_node_cluster_auto_slurm.py
@@ -76,12 +76,11 @@ def main(hparams, cluster):
     # ------------------------
     # 4 INIT TRAINER
     # ------------------------
-    gpus = list(range(0, hparams.per_experiment_nb_gpus))
     trainer = Trainer(
         experiment=exp,
         checkpoint_callback=checkpoint,
         early_stop_callback=early_stop,
-        gpus=gpus,
+        gpus=hparams.per_experiment_nb_gpus,
         nb_gpu_nodes=hyperparams.nb_gpu_nodes
     )