ubuntu下tensorflow的環(huán)境搭建
- 官網(wǎng)參考文檔地址: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
- 根據(jù)自己的情況選擇以下命令之一進行安裝:
$ 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包梗脾,不通的操作系統(tǒng)荸型、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/