Tensorflow 指令加速
生活随笔
收集整理的這篇文章主要介紹了
Tensorflow 指令加速
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
一直沒注意過使用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
總結
以上是生活随笔為你收集整理的Tensorflow 指令加速的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: [Writeup]与佛论禅
- 下一篇: 颜色协调模型Color Harmoniz