TensorFlow代码运行时出现AVX警告的原因及解决办法

tensorflow 代码执行时出现AVX警告的原因及解决办法:
2020-10-08 21:35:34.509010: W c:\tf_jenkins\home\workspace\release-win\m\windows\py\36\tensorflow\core\platform\cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations.
2020-10-08 21:35:34.509290: W c:\tf_jenkins\home\workspace\release-win\m\windows\py\36\tensorflow\core\platform\cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX2 instructions, but these are available on your machine and could speed up CPU computations.
原因:

这个警告意思是安装的TensorFlow支持AVX/AVX2,但是没有编译,使用AVX/AVX2可以提升CPU的执行效率。一般出现这种情况原因是安装TensorFlow时使用的是pip install tensorflow而不是源码安装。

解决办法:

在代码中添加两行代码指定日志级别,可屏蔽警告信息:

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值