ubuntu下tensorflow的环境搭建

左手的ㄟ右手 2022-04-13 04:29 328阅读 0赞

## ubuntu下tensorflow的环境搭建 ##

*  官网参考文档地址：[https://www.tensorflow.org/install/][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 -V`或`pip3 -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][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.tensorflow.org_get_started]

[https_www.tensorflow.org_install]: https://www.tensorflow.org/install/
[https_storage.googleapis.com_tensorflow_linux_cpu_tensorflow-1.1.0-cp34-cp34m-linux_x86_64.whl]: https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.1.0-cp34-cp34m-linux_x86_64.whl
[https_www.tensorflow.org_get_started]: https://www.tensorflow.org/get_started/