无root权限下的TensorFlow源码安装

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

  1. github下载tensorflow源码,在服务器中建立一个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,可是我明明已经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
  1. 打开python,测试import tensorflow as tf是否可以成功。
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容