1、首先查看cpu支持的指令集方法:

cat /pro/cpuinfo

如图:

说明,cpu仅仅支持avx2指令集,而tensorflow版本需要avx指令集的支持。avx指令集是和cpu硬件想绑定的。 当前avx512指令集只有intel对应的高级cpu在支持。

各个tensorflow对应的avx2版本下载渠道:GitHub - fo40225/tensorflow-windows-wheel: Tensorflow prebuilt binary for Windows

注意:仅仅对windows环境支持,对于linux环境暂时未找到办法,如果有谁有解决办法的,欢迎留言 

另外也可以自己选择构建:构建方式,GitHub - lakshayg/tensorflow-build: TensorFlow binaries supporting AVX, FMA, SSE

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐