From 353bcdf80f34a59ce22a09802847dac428a739b3 Mon Sep 17 00:00:00 2001
From: Colombo <lepersorium@gmail.com>
Date: Wed, 2 Oct 2019 20:17:56 +0400
Subject: [PATCH] cuda_cache_max_size

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

diff --git a/nnlib/nnlib.py b/nnlib/nnlib.py
index c8031bf..f2bd9ab 100644
--- a/nnlib/nnlib.py
+++ b/nnlib/nnlib.py
@@ -139,6 +139,8 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
 
         if 'CUDA_VISIBLE_DEVICES' in os.environ.keys():
             os.environ.pop('CUDA_VISIBLE_DEVICES')
+            
+        os.environ['CUDA_​CACHE_​MAXSIZE'] = '536870912' #512Mb (32mb default)
 
         os.environ['TF_MIN_GPU_MULTIPROCESSOR_COUNT'] = '2'
         os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' #tf log errors only