参考:
https://codegists.com/code/install-tensorflow-without-root/
https://www.tensorflow.org/install/install_sources
- 从github下载tensorflow源码,在服务器中建立一个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,可是我明明已经yes了cuda的使用了。原因在tensorflow/third_party/gpus下的cuda_configure.bzl的enable_cuda没能置1。根据stackoverflow一位网友的方案,我直接在.bashrc中 export TF_NEED_CUDA=1,问题就解决了!接下来继续:
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是否可以成功。