upd readme

This commit is contained in:
iperov 2019-01-12 15:29:34 +04:00
parent da0af0f35e
commit 81e837d920
2 changed files with 7 additions and 4 deletions

View file

@ -129,6 +129,9 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
if nnlib.tf is not None:
return nnlib.code_import_tf
if not nnlib.device.hasNVML():
print ("nvml.dll not found. Reinstall Geforce video drivers.")
if 'TF_SUPPRESS_STD' in os.environ.keys() and os.environ['TF_SUPPRESS_STD'] == '1':
suppressor = std_utils.suppress_stdout_stderr().__enter__()
else: