From a03f96fd0df0e2b57ead4a21c7e7fb9b82ee6bbe Mon Sep 17 00:00:00 2001
From: Priya Kasimbeg <kasimbeg@google.com>
Date: Mon, 25 Sep 2023 20:21:00 +0000
Subject: [PATCH] single_gpu

---
 submission_runner.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/submission_runner.py b/submission_runner.py
index 1e94c63e4..fcb4ac123 100644
--- 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