CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix for tf-directml

Browse Source
This commit is contained in:
iperov
2022-05-04 16:41:17 +04:00
parent b9e571d7e1
commit 0d19d8ec8e
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/leras/device.py
+1
View File
@@ -161,6 +161,7 @@ class Devices(object):
if 'CUDA_VISIBLE_DEVICES' in os.environ.keys(): if 'CUDA_VISIBLE_DEVICES' in os.environ.keys():
os.environ.pop('CUDA_VISIBLE_DEVICES') os.environ.pop('CUDA_VISIBLE_DEVICES')
os.environ['TF_DIRECTML_KERNEL_CACHE_SIZE'] = '2500'
os.environ['CUDA_​CACHE_​MAXSIZE'] = '2147483647' os.environ['CUDA_​CACHE_​MAXSIZE'] = '2147483647'
os.environ['TF_MIN_GPU_MULTIPROCESSOR_COUNT'] = '2' os.environ['TF_MIN_GPU_MULTIPROCESSOR_COUNT'] = '2'
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' # tf log errors only os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' # tf log errors only