python中用于分配tensorflow gpu内存的包
mute-tf-warnings的Python项目详细描述
静音TensorFlow警告
是否对tensorflow升级/其他警告感到沮丧?那么这个包裹是给你的。
它使tensorflow警告静音,并使代码输出干净。
安装
$ pip install mute_tf_warnings
用法
importtensorflowastffrommute_tf_warningsimporttf_mute_warning# run this before any other tensorflow codetf_mute_warning()