set CUDA_​CACHE_​MAXSIZE to 2gb before first cuda load

This commit is contained in:
iperov committed 2020-12-28 18:37:29 +04:00
1 parent 637c2b2a9b
commit cc950f12fb
1 file changed
+1
+1
View File
@@ -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: