update devicelib/nnlib to detect compute capability

2019-01-08 10:48:06 +04:00
parent 29c2375f5a
commit e8620919a7
5 changed files with 70 additions and 21 deletions
@@ -131,8 +131,8 @@ class ConvertSubprocessor(SubprocessorBase):
        from nnlib import nnlib          
        #model process ate all GPU mem,
        #so we cannot use GPU for any TF operations in converter processes (for example image_utils.TFLabConverter)
-        #therefore forcing prefer_DeviceConfig to CPU only
-        nnlib.prefer_DeviceConfig = nnlib.DeviceConfig (cpu_only=True)
+        #therefore forcing active_DeviceConfig to CPU only
+        nnlib.active_DeviceConfig = nnlib.DeviceConfig (cpu_only=True)
        
        return None