使用 Virtualenv 进行安装TensorFlow

使用 Virtualenv 进行安装TensorFlow

环境:ubuntu 16.04

  • 安装 pip 和 Virtualenv

    1
    2
    $  sudo apt-get install python-pip python-dev python-virtualenv # for Python 2.7
    $ sudo apt-get install python3-pip python3-dev python-virtualenv # for Python 3.n
  • 创建 Virtualenv 环境

  • 1
    2
    $  virtualenv --system-site-packages targetDirectory # for Python 2.7
    $ virtualenv --system-site-packages -p python3 targetDirectory # for Python 3.n

    targetDirectory 用于指定 Virtualenv 目录。我们假定创建的虚拟环境为tensorflow,即targetDirectorytensorflow

  • 激活 Virtualenv 环境

  • 1
    $  source ~/tensorflow/bin/activate

    执行上述 source 命令后,您的提示符应该会变成如下内容:

    1
    (tensorflow)$
  • 安装或更新pip

  • 1
    (tensorflow)$ easy_install -U pip

    安装 TensorFlow

    1
    2
    3
    4
    (tensorflow)$ pip install --upgrade tensorflow      # for Python 2.7
    (tensorflow)$ pip3 install --upgrade tensorflow # for Python 3.n
    (tensorflow)$ pip install --upgrade tensorflow-gpu # for Python 2.7 and GPU
    (tensorflow)$ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU
  • 备注:如果安装失败或者网速慢,可参考TensorFlow-CPU/GPU安装ubuntu16.04版 ,离线安装tensorflow

  • 退出 Virtualenv 环境

  • 1
    (tensorflow)$ deactivate

验证环境

  • 激活 Virtualenv 环境

  • 1
    $ source ~/tensorflow/bin/activate
    1
    2
    3
    4

    在 Virtualenv 环境激活后,您就可以从这个 shell 运行 TensorFlow 程序。您的提示符将变成如下所示,这表示您的 Tensorflow 环境已处于活动状态:

    -

    (tensorflow)$

    1
    2

    输入代码验证

    import tensorflow as tf

    hello = tf.constant(‘Hello, TensorFlow!’)

    sess = tf.Session()

    print(sess.run(hello))

    1
    2
    3
    4

    - 看到如下输出,表示安装正确

    -

    Hello, TensorFlow!

    1
    2

    用完 TensorFlow 后,可以通过发出以下命令调用 `deactivate` 函数来停用环境:

    (tensorflow)$ deactivate

    1
    2
    3
    4

    - 提示符将恢复为您的默认提示符 ,就表示退出了。

    -

    $

    1
    2
    3
    4
    5
    6
    7




    ### 卸载 Virtualenv 环境 `targetDirectory`

    要卸载你刚刚新建的Virtualenv 环境 `TensorFlow`的话,只需要删除该文件夹就可以了。

$ rm -r targetDirectory # 在这里 targetDirectory改为你创建的环境名字
`

转载请注明:Seven的博客

本文标题:使用 Virtualenv 进行安装TensorFlow

文章作者:Seven

发布时间:2018年08月11日 - 00:00:00

最后更新:2018年12月11日 - 22:03:03

原始链接:http://yoursite.com/2018/08/11/2018-08-11-TensorFlow-Virtualenv/

许可协议: 署名-非商业性使用-禁止演绎 4.0 国际 转载请保留原文链接及作者。

------ 本文结束------
坚持原创技术分享,您的支持将鼓励我继续创作!
0%