single_gpu

mlcommons · Sep 25, 2023 · a03f96f · a03f96f
1 parent 59c08ab
commit a03f96f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/submission_runner.py b/submission_runner.py
@@ -13,6 +13,8 @@
     --experiment_dir=/home/znado/experiment_dir \
     --experiment_name=baseline
 """
+import os
+os.environ["CUDA_VISIBLE_DEVICES"]='0'
 
 import datetime
 import gc