參考:
https://codegists.com/code/install-tensorflow-without-root/
https://www.tensorflow.org/install/install_sources
- 從github下載tensorflow源碼,在服務(wù)器中建立一個workspace目錄,把源碼放到這個目錄下面。
- 從這里下載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
- 在tensorflow下 ./configure,下載必要的依賴等。注意cuda項要yes, version -> 7.5。
- 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
- 打開python,測試import tensorflow as tf是否可以成功。