Ubuntu 16.04 bazel編譯tensorflow-gpu隨記

02-09

安裝Ubuntu16.04系統，以python2.7為例，安裝tensorflow GPU版本。

安裝cuda8.0版本後，下載cudnn，拷貝到cuda8.0相應的子目錄中，安裝命令

sudo pip install tensorflow-gpu

但會出現以下警告：

意思是說CPU支持SSE，AVX指令集加速，但安裝的tensorflow不支持，浪費資源潛力。

需要用bazel對tensorflow源碼編譯再安裝：

1. 安裝bazel，git。

2. 卸載已安裝的tensorflow：

sudo pip uninstall tensorflow-gpu

3. 克隆tensorflow源碼：

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

4. 進入tensorflow文件夾，運行配置文件：

cd tensorflow/

sudo ./configure

選擇支持cuda編譯，制定版本，可以不支持OPENCL編譯。

5. 源碼編譯，生成build_pip_package腳本，耗時半小時左右：

sudo bazel build -c opt --copt=-msse3 --copt=-msse4.1 --copt=-msse4.2 --copt=-mavx --copt=-mavx2 --copt=-mfma --config=cuda //tensorflow/tools/pip_package:build_pip_package

6. 在/tmp/tensorflow_pkg文件夾中創建pip的安裝包：

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

7. 安裝編譯後的tensorflow gpu版本：

sudo pip install /tmp/tensorflow_pkg/tensorflow-1.1.0rc1-cp27-cp27mu-linux_x86_64.whl

不再出現以上Warning!