1. 程式人生 > >ubuntu下tensorflow的環境搭建

ubuntu下tensorflow的環境搭建

ubuntu下tensorflow的環境搭建

  • 官網參考文件地址: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 -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/