美文网首页我爱编程程序员
TensorFlow安装(借助virtualenv)

TensorFlow安装(借助virtualenv)

作者: 吃土的汉子 | 来源:发表于2017-06-14 22:25 被阅读0次

    virtualenv可以为python配置不同的环境(包依赖,python版本等), 借助virtualenv安装tensorflow,可以方便使用不同的python版本(例如python2, python3), 也可以方便的卸载(直接将env删除)

    安装

    前置条件:安装了python/virtualenv/python-pip

    1. 为tensorflow创建一个env
      virtualenv tensorflow
      目录下将出现一个叫tensorflow的目录,目录下bin/lib等一应俱全,这就是virtualenv为我们准备的单独环境
    2. 切换到该环境
      source tensorflow/bin/activate
    3. 安装tensorflow
      pip install --upgrade tensorflow
      稍等片刻, tensorflow即安装成功了.

    Hello World

    尝试运行以下Python代码:
    <code>
    import tensorflow as tf

    hello = tf.constant('Hello, TensorFlow!')

    session = tf.Session()
    print session.run(hello)
    </code>

    你可能会遇到类似以下警告:
    W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.

    这是由于直接用pip安装的是兼容各种cpu的版本, 并不支持你的CPU的一些高级特性. 如果你想提高性能,需要从源码编译

    这条警告可以通过加入以下代码移除:
    <code>
    import os
    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
    </code>

    卸载

    将virtualenv为你创建的目录删除即可
    rm -rf tensorflow

    相关文章

      网友评论

        本文标题:TensorFlow安装(借助virtualenv)

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