ubuntu下tensorflow的环境搭建

前端之家收集整理的这篇文章主要介绍了ubuntu下tensorflow的环境搭建前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

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/

原文链接:https://www.f2er.com/ubuntu/353247.html

猜你在找的Ubuntu相关文章