Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

moved cuda flags inside trainer

This commit is contained in:
William Falcon 2019-07-08 09:58:11 -04:00
parent f5a87c5016
commit 2172280d8d
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/new_project_templates/trainer_gpu_cluster_template.py
View File

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