1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > 完美解决Tensorflow不支持AVX2指令集问题|指令集加速

完美解决Tensorflow不支持AVX2指令集问题|指令集加速

时间:2024-04-05 23:14:25

相关推荐

完美解决Tensorflow不支持AVX2指令集问题|指令集加速

在pycharm中安装tensorflow后

运行如下测试代码:

import tensorflow as tfx = tf.Variable(3, name="x")y = tf.Variable(4, name="y")f = x*x*y + y + 2print(f)

发现会报一行错误

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2大概意思是安装的tensorflow版本不支持cpu的AVX2编译

可能是因为安装时使用的pip install tensorflow ,这样默认会下载X86_64的SIMD版本。

查找解决办法后,有以下两种办法:

1.忽略屏蔽这个警告

在代码最前面添加如下两行代码

import osos.environ["TF_CPP_MIN_LOG_LEVEL"]='2' # 只显示 warning 和 Error

2.彻底解决,换成支持cpu用AVX2编译的TensorFlow版本。

首先卸载原来安装的tensorflow版本

pip uninstall tensorflow

在这里下载对应版本的tensorflow:/fo40225/tensorflow-windows-wheel,比如我需要的是CPU+AVX2+Python3.6,那么我就在下面的列表中选择这个:

找到对应的.whl文件

下载该文件,我用google浏览器下载一直显示无法访问

后来选用Edge浏览器打开就好啦,直接就下载成功了。

此处放上tensorflow-1.12.0-cp36-cp36m-win_amd64.whl的下载链接:

链接:/s/1CvKUtmM1zHyJyJk87eFEUA

提取码:o85f

然后用activate 进入自己创建的虚拟环境

运行pip install tensorflow-1.12.0-cp36-cp36m-win_amd64.whl命令安装

最后用conda list命令看安装了那些包

然后再次运行代码,就不会报AVX2的错误啦

参考链接:/beyond9305/article/details/95896135/article/179405.htm

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。