From cc950f12fb951ad13d410797ba4c3b9024245c0d Mon Sep 17 00:00:00 2001 From: iperov Date: Mon, 28 Dec 2020 18:37:29 +0400 Subject: [PATCH] =?UTF-8?q?set=20CUDA=5F=E2=80=8BCACHE=5F=E2=80=8BMAXSIZE?= =?UTF-8?q?=20to=202gb=20before=20first=20cuda=20load?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- core/leras/device.py | 1 + 1 file changed, 1 insertion(+) diff --git a/core/leras/device.py b/core/leras/device.py index 5e6729d..4d157f0 100644 --- a/core/leras/device.py +++ b/core/leras/device.py @@ -87,6 +87,7 @@ class Devices(object): os.environ['NN_DEVICES_INITIALIZED'] = '1' os.environ['NN_DEVICES_COUNT'] = '0' + os.environ['CUDA_​CACHE_​MAXSIZE'] = '2147483647' min_cc = int(os.environ.get("TF_MIN_REQ_CAP", 35)) libnames = ('libcuda.so', 'libcuda.dylib', 'nvcuda.dll') for libname in libnames: