1. 程式人生 > >TensorFlow:Ubuntu 下安裝 TensorFlow

TensorFlow:Ubuntu 下安裝 TensorFlow

最近為了學習 tensorflow,要在主機上安裝 tensorflow。所以在 github 上面下載了一個官方編譯的 whl 來安裝。

一、主機

主機為雙路的 Titan xp,系統是 Ubuntu 16.04

二、配置過程

(一)先更新顯示卡驅動

“系統設定”-- “軟體與更新”-- “額外驅動”--選擇第一個就好了。

選好後點應用,輸入密碼,就會自動開始下載更新。

(二)安裝CUDA9.0

網址:

根據自己的系統選擇 runfile,儲存檔案。

下載完成後,進入到下載目錄,(我的下載在  ~/Downloads/  裡)。

#給檔案新增執行許可權

chmod +x ./cuda_9.0.176_384.81_linux.run

#執行安裝

sudo ./cuda_9.0.176_384.81_linux.run

(這裡會出現一大篇文字,長按 <Enter> or <Ctrl + c> 跳過 ,到最後打上 accept)

注意:第一個提醒你是否安裝驅動時,選“n”

其餘都“y”

#新增環境

gedit ~/.bashrc

把下面的內容新增到最後:

export CUDA_HOME=/usr/local/cuda

export PATH=$PATH:$CUDA_HOME/bin

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

#測試是否安裝成功

nvcc --version

若輸出

則代表安裝成功

(三)下載 cudnn 7.0.5

註冊個賬號,就可以下載了。

下載完解壓,裡面有個cuda檔案,裡面有兩個檔案include 和 lib64 ,把裡面的檔案copy到/usr/local/cuda/裡面相應的目錄裡。

#sudo cp ~/Downloads/cuda/include/cudnn.h /usr/local/cuda/include/

#sudo cp ~/Downloads/cuda/lib64/libcudnn* /usr/local/cuda/lib64/

給檔案加讀許可權

#sudo chmod a+x /usr/local/cuda/include/cudnn.h

#sudo chmod a+x /usr/local/cuda/lib64/libcudnn*

(四)下載官方編譯的whl檔案

往下拉,在

第二行的build history裡,找到1.8.0版本的whl包下載

(五)pip 安裝whl檔案

在 .whl 檔案位置開啟終端,

sudo pip install tf_nightly_gpu-1.8.0.dev20180416-cp27-cp27mu-linux_x86_64.whl

(六)測試是否安裝完成

python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> sess.run(hello)
'Hello, TensorFlow!'

沒有報錯則安裝成功

(七)檢視tensorflow版本

>>> tf.__version__
'1.8.0-dev20180416'

基本上按照這個方法安裝沒什麼問題,如果有什麼問題,歡迎給我留言。