編譯TensorFlow CPU指令集優化版


 

編譯TensorFlow CPU指令集優化版

如題,CPU指令集優化版,說的是針對某種特定的CPU型號進行過優化的版本。通常官方給的版本是沒有針對特定CPU進行過優化的,有網友稱,優化過的版本相比優化前的版本性能提升大概30%。

下面簡單介紹下在Ubuntu上進行Tensor Flow編譯

必要的環境:

  1. bazel, 這玩意的安裝方法參考別的資料,比較麻煩
  2. sudo apt-get install python-numpy python-dev python-pip python-wheel
  3. 因為tf的版本與python版本是有關的,所以如果你有建立虛擬python環境,比如你用conda創建了一個虛擬的tf環境‘tensorflow’,那么通過source activate tensorflow先進入該環境,此時顯示(tensorflow) lev@lev-Precision-Tower-5810:~/類似的
  4. gcc 4.7+

下面介紹具體的編譯工作:

  1. 在你的工作目錄下,執行git clone https://github.com/tensorflow/tensorflow.git
  2. 進入tensorflow目錄,cd tensorflow
  3. 開始配置編譯參數,./configure,根據需要去配置好對應的東東
  4. 接下來進入編譯,bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package, 采用默認優化,該設定會自動編譯成你的cpu所支持的指令集版本
  5. 編譯whl,也就是python的package,bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg,此操作會在將whl生成在/tmp/tensorflow_pkg目錄下
  6. 安裝whl, 去/tmp/tensorflow_pkg下找到你的造的輪子,比如我的pip install /tmp/tensorflow_pkg/tensorflow-1.4.0-cp36-cp36m-linux_x86_64.whl, 這個時候,記得退出tensorflow的源碼目錄,再用命令行運行python,導入tf,如果不退出,直接運行python環境import tensorflow會報錯,因為此時import的是源碼目錄里的tensorflow, 不是你剛剛安裝好的。

好了,現在你可以進行愉快的玩耍了,have fun!

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM