From e692c079ed2b8611ce1b232c5881e09803cf4887 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Tue, 26 Sep 2023 01:12:18 +0000 Subject: [PATCH] reduce splitsize --- submission_runner.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/submission_runner.py b/submission_runner.py index e2d37cb3f..626fb2c0b 100644 --- a/submission_runner.py +++ b/submission_runner.py @@ -15,6 +15,8 @@ """ import os # os.environ["CUDA_VISIBLE_DEVICES"]='0' +os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:256" + import datetime import gc