1. 程式人生 > 實用技巧 >安裝TensorFlow 完整教程 GPU版 2020-10-7

安裝TensorFlow 完整教程 GPU版 2020-10-7

(我下載的GPU版本 版本號為1.15.2 先說下我的顯示卡 是GTX 960M )

第一步:去清華大學映象網站下載 Anaconda

網址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

我下載的是這個

第二步:安裝Anaconda 就按預設的安裝路徑就完了 別多此一舉換路徑 太煩了 要不

第三步:驗證一下安裝的python 版本

開啟

輸入 python 會得到如下圖

第四步:更換PIP源地址 (這步很重要 不換的話 你就用你那個十幾kb去下載吧 慢死你 換完以後 能達到6MB/S的速度 很快滴 )

教程網址:

http://www.ipaomi.com/2017/11/04/python-pip-%E4%B8%8B%E8%BD%BD%E9%80%9F%E5%BA%A6%E6%85%A2-windows-%E8%AE%BE%E7%BD%AE-%E5%9B%BD%E5%86%85%E6%BA%90%EF%BC%8C%E7%94%A8-%E9%98%BF%E9%87%8C%E4%BA%91-%E5%9B%BD%E5%86%85%E9%95%9C%E5%83%8F/

簡單說一下:1.先在 windows “檔案資源管理器” 位址列 輸入%APPDATA% 按回車,新建資料夾 命名為pip

2.在剛才建立好的 pip 資料夾中,新建 名為 pip.ini 的配置檔案

3.在 pip.ini 檔案中輸入以下內容,然後儲存

[global]

index-url=http://mirrors.aliyun.com/pypi/simple/

[install]

trusted-host=mirrors.aliyun.com

完成!!! ^ . ^

第五步:去下載pycharm (下載專業版就好 那個學生能免費,如果你有教育郵箱 直接用你的教育郵箱就行 能獲得資格; 如果沒有 看下面的這個連結 需要你去學信網下載個學籍證明 發過去 之後他會給你回郵件告訴你獲得了資格

https://blog.csdn.net/qq_44799719/article/details/108032182?utm_medium=distribute.pc_relevant.none-task-blog-OPENSEARCH-3.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-OPENSEARCH-3.nonecase

這個安裝步驟不用說了吧 就那麼裝

第六步:安裝CUDA(10.0) 和 CuDNN(7.5.0) 版本在括號裡面 不要下載錯了 這個很重要 一定要對應 下面貼網址

CUDA:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64

TIPS:安裝的時候注意一下,他有大概四個可以勾選的吧 你看他的版本是否高於你當前的版本,如果高於 就勾選上,如果低於就別選, 安裝還是預設路徑

CuDNN:https://developer.nvidia.com/rdp/cudnn-archive

這個下載需要你登入你的nvidia賬號 登陸完就可以下載了

下載完了解壓縮到CUDA的安裝目錄就行了 預設是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

小插曲:官網上顯示是CuDNN7.4 但是我按照清華大學出版的一本書 下載的CuDNN7.5.0 目前也用的沒問題 沒報錯 所以 應該沒問題吧

配置CUDA環境變數 我加了三個分別為:

1.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

2.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64

3.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include

第七步:安裝TensorFlow gpu 最後一步!!!

開啟

輸入: python -m pip install --user tensorflow-GPU==1.15.2

然後等著安裝完就行了,這時候前面第四步就起作用了 6-8MB/s的速度美滋滋 之前我沒設定 他的速度是15kb/s 真的是氣死了 媽的

解釋一下:

python -m 加這個是因為我的pip為最高版本 所以加一個讓他適應當前版本下載包 不然可能下載不了

--user 加這個是因為 CUDA預設按在了使用者那裡 所以可能沒有管理員許可權 你加了這個語句 就有了管理員許可權 可以下載讀寫檔案了

最後:如果你在下載過程中遇到什麼問題,仔細看一下他報錯在哪裡,然後ctrl+c複製到百度上 基本都能搜出解決方法 ,然後你按著他的方法試一下基本就解決了 還有 你可以去圖書館借相關書籍 這是個非常靠譜的做法,我就是因為習慣性的喜歡網路上找教程,但是無奈他們的方法對應版本都不一樣,出現的問題都亂七八糟的 所以我去圖書館借了本清華出的TensorFlow教程 按著上面的步驟 很快就裝好了 雖然中間出了點問題 但是基本都能百度自己解決