Tensorflow不支持不支持AVX2指令集的解决方法指令集的解决方法
今天小编就为大家分享一篇Tensorflow不支持AVX2指令集的解决方法,具有很好的参考价值,希望对大家有所
帮助。一起跟随小编过来看看吧
这几天研究了一下FCN(全卷积网络),由于电脑配置不够,用GPU训练直接报OOM(内存溢出)了, 于是转战CPU,当然,
这样会很慢,之后会继续搞一下,减小一下网络的复杂度,对一些参数设置一波,看能不能正常跑下来。
记得一开始没有装GPU版的tensorflow时用CPU版本跑程序的时候总是报警告:Your CPU supports instructions that this
TensorFlow binary was not compiled to use: AVX AVX2,当时没有太在意,大概搜了一下,就是说你的电脑是支持AVX2
的,但是你装的tensorflow版本却不支持,当然,如果有GPU的话就可以忽视这个了,毕竟优先使用后者~既然可以更好一点
(使用AVX2),那就追求一下完美吧……
网上很大部分资料对于这一块采取的措施都是屏蔽掉,所谓眼不见心不烦,但这样只能是自欺欺人啊,治标不治本,于是探究
了一下,成功解决了这一问题,现记录一下:
1. 首先在Anaconda中创建虚拟环境,命名为cpu_avx2,python版本指定为3.7,这样避免出错崩溃影响到其他程序操作:
2. 在这里下载对应版本的tensorflow:https://github.com/fo40225/tensorflow-windows-wheel,比如我需要的是
CPU+AVX2+Python3.7,那么我就选择第二个:
按照路径提示,在上边找到对应的.whl文件,对应本次安装的路径为:
3. 将该文件下载下来,放到相应的目录下(为了方便建议放到命令窗口对应的目录下,这样就不用再切换目录了,反正只要
能找到该文件就行):
然后打开Anaconda Prompt,进入刚才创建的虚拟空间cpu_avx2,安装即可:
评论0
最新资源