From 8ad2a5373e810e2a7eeac3e896373748f638f2de Mon Sep 17 00:00:00 2001 From: Colombo Date: Tue, 24 Mar 2020 22:06:10 +0400 Subject: [PATCH] fix Exception("nn devices are not initialized. Run initialize_main_env() in main process.") --- core/leras/device.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/core/leras/device.py b/core/leras/device.py index 46fbd12..5e6729d 100644 --- a/core/leras/device.py +++ b/core/leras/device.py @@ -84,6 +84,9 @@ class Devices(object): @staticmethod 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)) libnames = ('libcuda.so', 'libcuda.dylib', 'nvcuda.dll') for libname in libnames: @@ -129,7 +132,6 @@ class Devices(object): }) cuda.cuCtxDetach(context) - os.environ['NN_DEVICES_INITIALIZED'] = '1' os.environ['NN_DEVICES_COUNT'] = str(len(devices)) for i, device in enumerate(devices): os.environ[f'NN_DEVICE_{i}_NAME'] = device['name']