美文网首页我爱编程
无root权限下的TensorFlow源码安装

无root权限下的TensorFlow源码安装

作者: 麦兜胖胖次 | 来源:发表于2017-03-22 19:07 被阅读0次

    参考:
    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是否可以成功。

    相关文章

      网友评论

        本文标题:无root权限下的TensorFlow源码安装

        本文链接:https://www.haomeiwen.com/subject/uxzonttx.html