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