無root權(quán)限下的TensorFlow源碼安裝

參考:
https://codegists.com/code/install-tensorflow-without-root/
https://www.tensorflow.org/install/install_sources

  1. github下載tensorflow源碼,在服務(wù)器中建立一個workspace目錄,把源碼放到這個目錄下面。
  2. 這里下載bazel,最好下載最新的,很老版本的可能不支持。直接下載installer版(.sh后綴)的,運行前chmod +x ×××.sh,之后./×××.sh安裝。注意如果說jdk版本過老,需要先去下載最新的jdk8,并且把.bashrc里面的JAVA_HOME和PATH更改或添加:
export JAVA_HOME=/home/user/jdk/jdk1.8.0_111
PATH=$JAVA_HOME/bin:$PATH
  1. 在tensorflow下 ./configure,下載必要的依賴等。注意cuda項要yes, version -> 7.5。
  2. bazel就像cmake一樣,方便了tf的安裝。
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

上面的命令行報錯說configure階段沒有指明使用gpu,可是我明明已經(jīng)yes了cuda的使用了。原因在tensorflow/third_party/gpus下的cuda_configure.bzl的enable_cuda沒能置1。根據(jù)stackoverflow一位網(wǎng)友的方案,我直接在.bashrc中 export TF_NEED_CUDA=1,問題就解決了!接下來繼續(xù):

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

sh最后pip install一下:

pip install /tmp/tensorflow_pkg/tensorflow-×××.whl --user
  1. 打開python,測試import tensorflow as tf是否可以成功。
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容