From 2172280d8d0970797104f9d2ccd8d1e6d40d071e Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Mon, 8 Jul 2019 09:58:11 -0400
Subject: [PATCH] moved cuda flags inside trainer

---
 examples/new_project_templates/trainer_gpu_cluster_template.py | 3 ---
 1 file changed, 3 deletions(-)

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
     # ---------------------