set CUDA_CACHE_MAXSIZE to 2gb before first cuda load

2020-12-28 18:37:29 +04:00
parent 637c2b2a9b
commit cc950f12fb
1 changed files with 1 additions and 0 deletions
@@ -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: