ubuntu編譯tensorflow 支援AVX,AVX2等指令集

1.下載下傳安裝bazel

$ sudo apt-get update && sudo apt-get install bazel
           

https://docs.bazel.build/versions/master/install-ubuntu.html

2.克隆代碼

$ git clone https://github.com/tensorflow/tensorflow
           

 如果想編譯某一個版本：以1.13版本為例

$ git clone -b r1.13 https://github.com/tensorflow/tensorflow
           

3.使用conda虛拟環境

conda create -n avx python=3.6
conda activate avx
pip install numpy
pip install six 
pip install wheel 
           

 4.編譯

$ cd tensorflow 
$ ./configure # 都選擇N
           

都選N，不影響AVX，AVX2等的編譯

5.bazel build

$ bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
           

6.生成whl檔案

bazel build

 指令會建立一個名為 

build_pip_package

 的可執行檔案，此檔案是用于編譯 

pip

 軟體包的程式。請如下所示地運作該可執行檔案，以在 

/tmp/tensorflow_pkg

 目錄中編譯 

.whl

 軟體包。

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

優化後的cpu運算速度可以快三到四倍。