一直没注意过使用Tensorflow的时候有一条warning:
Warning: your cpu supports instructions that this tensorflow binary was not compiled to use: avx2 fma
这玩意是可以用来加速推断的,分CPU和GPU版,业务相关部署在CPU上,实测使用CPU加速能提升3倍左右,但是编译过程有点复杂。所以在本地编译了一个Ubuntu16.04 Python3版本的支持AVX2和FMA指令加速的tensorflow。分享一下编译包,以及编译完以后的whl安装包
【国际惯例】
编译时参考的博客:
通过源码编译安装TensorFlow-CPU版本支持AVX等指令集
分享:
编译过程包:
链接: https://pan.baidu.com/s/1ROK1hQccTSwwkYzPW6isPQ 提取码: gt9c
编译完毕的支持AVX2 FMA XLA的tensorflow包
链接: https://pan.baidu.com/s/1qCiuXihevdwccJcNQ3f7Fg 提取码: jj2e