美文网首页Python我爱编程
ubuntu下tensorflow的环境搭建

ubuntu下tensorflow的环境搭建

作者: 一叶一声秋 | 来源:发表于2017-04-27 15:36 被阅读912次

    ubuntu下tensorflow的环境搭建

    • 官网参考文档地址:https://www.tensorflow.org/install/,由于博主用的是ubuntu 14.04,因此以Ubuntu为例,进行说明,其他OS请参考上面给出的链接。

    提示:最好别用windows,后面会有很多的兼容问题

    • 安装的方式也有好几种,通过pip,docker,Anacodnda等,因为ubuntu是自带Python和pip的,因此这里给出的是pip的安装方式。

    确定python及pip的版本

    1. 输入命令python -V确认python的版本,需要2.7或者是3.3+

    2. 输入命令pip -Vpip3 -V确认pip的版本,建议pip在8.1以上,或者是pip3,如果不是则使用sudo apt-get install python-pip python-dev进行更新。

    安装tensorflow

    1. 根据自己的情况选择以下命令之一进行安装:
     $ pip install tensorflow      # Python 2.7; 仅支持CPU
     $ pip3 install tensorflow     # Python 3.n; 仅支持CPU
     $ pip install tensorflow-gpu  # Python 2.7; 支持CPU
     $ pip3 install tensorflow-gpu # Python 3.n; 支持CPU 
    
    1. 该步骤为可选步骤,如果上一步失败了,可以通过以下命令来安装:
    $ sudo pip  install --upgrade TF_PYTHON_URL   # Python 2.7
    $ sudo pip3 install --upgrade TF_PYTHON_URL   # Python 3.N 
    

    其中,TF_PYTHON_URL为TensrorFlow的python包,不通的操作系统、python版本、GPU支持状况需要选择不同的包,例如OS为Linux,python版本为3.4,仅支持CPU的情况下,TF_PYTHON_URL应当替换为 https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.1.0-cp34-cp34m-linux_x86_64.whl

    验证tensorflow是否安装成功

    1. 启动终端,输入python
    2. 输入以下代码:
    >>> import tensorflow as tf
    >>> hello = tf.constant('Hello, TensorFlow!')
    >>> sess = tf.Session()
    >>> print(sess.run(hello))
    

    如果输出Hello, TensorFlow!则代表安装成功。
    想要了解更多的tensorflow信息,可以直接访问https://www.tensorflow.org/get_started/

    相关文章

      网友评论

        本文标题:ubuntu下tensorflow的环境搭建

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