ubuntu下tensorflow的环境搭建
- 官网参考文档地址:https://www.tensorflow.org/install/,由于博主用的是ubuntu 14.04,因此以Ubuntu为例,进行说明,其他OS请参考上面给出的链接。
提示:最好别用windows,后面会有很多的兼容问题
- 安装的方式也有好几种,通过pip,docker,Anacodnda等,因为ubuntu是自带Python和pip的,因此这里给出的是pip的安装方式。
确定python及pip的版本
-
输入命令
python -V
确认python的版本,需要2.7或者是3.3+ -
输入命令
pip -V
或pip3 -V
确认pip的版本,建议pip在8.1以上,或者是pip3,如果不是则使用sudo apt-get install python-pip python-dev
进行更新。
安装tensorflow
- 根据自己的情况选择以下命令之一进行安装:
$ 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
- 该步骤为可选步骤,如果上一步失败了,可以通过以下命令来安装:
$ 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是否安装成功
- 启动终端,输入
python
- 输入以下代码:
>>> 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/
网友评论