Bootstrap

TensorFlow SSE/AVX/FMA 指令集编译

当TensorFlow出现

I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA 
等警告时,表明你的CPU指令集包括了这些指令集,当用pip的方式安装默认是不编译这些的,如果需要编译这些以提高效率,需要自己下载tensorflow的源码,进行编译。

具体的编译方法网上有不少文章,但中文的资料多少都有些不完整,如下面一篇博客算是比较完整的了

https://blog.csdn.net/iTaacy/article/details/72799833

但第5步之前仍然少了一个步骤去生成.wheel文件,生成.wheel文件的方法如下:

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

参考自:

https://stackoverflow.com/questions/41293077/how-to-compile-tensorflow-with-sse4-2-and-avx-instructions

这样才算完整的步骤。

 

值得一提的是,虽然解决了开头说的警告,但在我用 的一个分词工具上,效率并没有提升。。分词工具的作者也联系不上。。

;