ubuntu下tensorflow的環境搭建
阿新 • • 發佈:2018-12-03
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/