Python我爱编程

ubuntu下tensorflow的环境搭建

2017-04-27  本文已影响912人  一叶一声秋

ubuntu下tensorflow的环境搭建

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

确定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/

上一篇下一篇

猜你喜欢

热点阅读