update devicelib/nnlib to detect compute capability

This commit is contained in:
iperov committed 2019-01-08 10:48:06 +04:00
1 parent 29c2375f5a
commit e8620919a7
5 files changed
+70 -21

No files matched your search

+2 -2
View File
@@ -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