1. 程式人生 > >TensorFlow-gpu安裝

TensorFlow-gpu安裝

這裡我整整拖了三個月,一個是學習Linux和課題壓力的原因,還有一個原因就是堅持很久的東西突然斷掉了就很不情願重新拾起,我要改掉這個惡習。

重要細節:

1.將國內映象源改成中科大而不是清華映象!!!
首先說明,anaconda預設的下載路徑是外國網址,下載速度很慢,首先將預設下載地址改成國內映象。
試了清華映象還是很慢,就是因為這個一直沒有頭緒,今天看到一個帖子說中科大的映象更好用更快!!!
2.下載Python3.5及以下版本否則報錯

開始

開啟Anaconda Prompt,進入Anaconda命令列管理介面。

conda config –add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/ 
conda config –set show_channel_urls yes

開啟C盤使用者下的.condarc檔案刪掉第三行如下:
在這裡插入圖片描述
檢視是否生效,通過命令 conda info 檢視當前配置資訊,內容如下,即修改成功,關注 channel URLs 欄位內容
在這裡插入圖片描述
隨後以管理員身份執行命令列,輸入:

conda create -n tensorflow-gpu python=3.5

然後輸入y,紅框中會被下載

升級pip到最新版,防止稍後的安裝時,出現錯誤,輸入指令:

python -m pip install --upgrade pip

安裝tensorflow及相應依賴包,輸入指令:

pip install --ignore-installed --upgrade tensorflow-gpu

參考TensorFlow社群

並不會成功,因為Python版本問題會出現異常報錯。將上面3.6改成3.5再來一遍就可以了。
驗證
在命令列中,進入python,並輸入以下程式碼:

mport tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

輸出:
b’ Hello, TensorFlow!
大功告成!!
雖然我的970T顯示卡只能跑一些簡單的程式,但是我也很滿意了。確實比CPU跑起來快不少。

山高月小,水落石出