github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 21:12:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
iperov 2020-12-28 18:37:29 +04:00
parent 637c2b2a9b
commit cc950f12fb
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
core/leras/device.py
View file

@ -87,6 +87,7 @@ class Devices(object):
os.environ['NN_DEVICES_INITIALIZED'] = '1' os.environ['NN_DEVICES_INITIALIZED'] = '1'
os.environ['NN_DEVICES_COUNT'] = '0' os.environ['NN_DEVICES_COUNT'] = '0'
os.environ['CUDA_​CACHE_​MAXSIZE'] = '2147483647'
min_cc = int(os.environ.get("TF_MIN_REQ_CAP", 35)) min_cc = int(os.environ.get("TF_MIN_REQ_CAP", 35))
libnames = ('libcuda.so', 'libcuda.dylib', 'nvcuda.dll') libnames = ('libcuda.so', 'libcuda.dylib', 'nvcuda.dll')
for libname in libnames: for libname in libnames: