如何在WSL2中关闭TensorFlow的NUMA节点读取错误提示
关闭WSL2中TensorFlow的NUMA错误提示
这个提示属于TensorFlow的INFO级日志,是因为WSL2内核默认未启用NUMA支持导致的,不影响TensorFlow的正常运行,以下是几种屏蔽该提示的方法:
方法1:在Python代码中临时设置
在你的TensorFlow代码开头添加以下代码,直接调整日志级别过滤INFO级提示:
import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
参数说明:
0:默认值,显示所有日志(INFO/WARNING/ERROR/FATAL)1:屏蔽INFO级日志2:屏蔽INFO和WARNING级日志3:屏蔽所有日志(包括ERROR)
这里设置为2即可屏蔽目标提示,同时保留ERROR级别的关键日志。
方法2:全局环境变量设置(永久生效)
如果不想每次修改代码,可在WSL2的shell配置文件中添加环境变量:
- 打开你的shell配置文件(bash用户编辑
~/.bashrc,zsh用户编辑~/.zshrc) - 添加一行:
export TF_CPP_MIN_LOG_LEVEL=2
- 执行以下命令使配置生效:
source ~/.bashrc # bash用户执行 # 或 source ~/.zshrc zsh用户执行
之后每次启动WSL2运行TensorFlow,该提示都会被自动屏蔽。
内容的提问来源于stack exchange,提问作者KWIN
相关产品推荐
相关产品推荐

