关闭tensorflow警告和日志的简单python包。
silence-tensorflow的Python项目详细描述
关闭tensorflow警告和日志的简单python包。
如何安装此软件包?
像往常一样,只需使用pip:
pip install silence_tensorflow
我怎么用?
您只需要在导入tensorflow之前导入包:
importsilence_tensorflowimporttensorflowastf...# your code
这是怎么在引擎盖下工作的?
此程序包将把kmp_affinity系统变量设置为“noverbose”,tf_cpp_min_log_level设置为level2(仅记录错误),并使未来警告和不推荐警告都保持沉默。
如果您需要为kmp_affinity自定义值,则应在导入包后重置该值,如下所示:
importosbackup=os.environ["KMP_AFFINITY"]importsilence_tensorflowos.environ["KMP_AFFINITY"]=backup