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/

猜你喜欢

转载自blog.csdn.net/Scythe666/article/details/84564080
今日推荐