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

fix Exception("nn devices are not initialized. Run initialize_main_env() in main process.")

Browse Source
This commit is contained in:
Colombo
2020-03-24 22:06:10 +04:00
parent 2d5e949100
commit 8ad2a5373e
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/leras/device.py
+3 -1
View File
@@ -84,6 +84,9 @@ class Devices(object):
@staticmethod @staticmethod
def initialize_main_env(): def initialize_main_env():
os.environ['NN_DEVICES_INITIALIZED'] = '1'
os.environ['NN_DEVICES_COUNT'] = '0'
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:
@@ -129,7 +132,6 @@ class Devices(object):
}) })
cuda.cuCtxDetach(context) cuda.cuCtxDetach(context)
os.environ['NN_DEVICES_INITIALIZED'] = '1'
os.environ['NN_DEVICES_COUNT'] = str(len(devices)) os.environ['NN_DEVICES_COUNT'] = str(len(devices))
for i, device in enumerate(devices): for i, device in enumerate(devices):
os.environ[f'NN_DEVICE_{i}_NAME'] = device['name'] os.environ[f'NN_DEVICE_{i}_NAME'] = device['name']