centos 7 tensorflow-gpu-1.8.0 源碼編譯

這里我們安裝tensorflow-gpu-1.8.0
0.檢查gcc,cuDNN,CUDA的版本信息
1. 安裝bazel
舊版本安裝

  • 下載源 https://github.com/bazelbuild/bazel/releases?after=0.12.0
  • sudo chmod +x bazel-0.9.0-installer-linux-x86_64.sh
  • sudo ./bazel-0.9.0-installer-linux-x86_64.sh --user
    加入系統(tǒng)
  • source /home/gai/.bazel/bin/bazel-complete.bash
  • export PATH=$PATH:/home/gai/.bazel/bin
    需要注意的是，把gai換成自己系統(tǒng)的用戶名
注意：我用0.9.0編譯發(fā)現報錯，github有人說要把bazel版本提高到12；[知乎]（https://zhuanlan.zhihu.com/p/39772050）上也有人采坑啦~

最新版本安裝
https://docs.bazel.build/versions/master/install-ubuntu.html

  • 添加源 將源vbatts-bazel-epel-7.repo放在/etc/yum.repos.d目錄下。
    -sudo yum install bazel
    如果網絡不好可以，多次嘗試
    因為安裝的時候自動安裝了很多依賴項，時間可能比較久


2.安裝 TensorFlow Python 依賴項
yum install numpy, python-devel,python-pip, python-wheel

3. tensorflow下載配置

  • 下載對應版本的源碼解壓
    https://github.com/tensorflow/tensorflow/releases

    在進行configure配置之前，先安裝clang

  • sudo yum install epel-release

  • sudo yum install clang
    再進行配置 ,在tensorflow目錄下(可以從github下載source或者直接git這個比較慢...) ./configure 進行配置

如果出現這個錯誤：https://github.com/tensorflow/tensorflow/issues/10718



類似這種@bazel_tools可能就是bazel版本不匹配。

編譯完成后執(zhí)行

  • bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
  • sudo pip install /tmp/tensorflow_pkg/tensorflow-1.8.0-cp36-cp36m-linux_x86_64.whl
    如果是python3.6 需要sudo pip, 那還是用全路徑（pip）不然的話sudo默認使用pip2就會報tf的版本不適用這個平臺的錯誤。
    注意：
    configure從gpu版本切換cpu都要進行
  • try 'bazel clean' before /.configure.
    測試的時候別在tensorflow目錄下import tensorflow，可能直接引用里面的目錄下的包。

http://www.alfrednanwu.com/machine-learning/-gtx-1080ubuntu1604cuda8cudnn51tensorflow

?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成，瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明：文章內容（如有圖片或視頻亦包括在內）由作者上傳并發(fā)布，文章內容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容