真的不得不說,一開始我看了網路上的影片發覺應該蠻簡單安裝的,但實際去實作發現安裝tensorflow一步都不能錯,而且還要考慮自己的硬體設備,起初真的讓我嘗盡了苦頭,那我就先跟大家說一下要安裝tensorflow一些基本的架構吧。
再說下載步驟前我先介紹AI三寶
1.GPU
首先最底層的就是GPU了,這裡我也簡單的說明一下,當前多核的CPU大多由4或6個核所組成,但普通的GPU就包含了幾百個核,這對於多媒體處理中大量的重複處理過程有著天生的優勢,同時更重要的是,它可以用來做大規模並行數據處理。因此GPU不僅在圖像處理中應用廣泛,還在科學計算、密碼破解、數值分析、大數據處理、金融分析等需要並行運算的領域中廣為應用,GPU是一種較為通用的專業晶片。
2.CUDA
是由NVIDIA所推出的一種整合技術,是該公司對於GPGPU的正式名稱。透過這個技術,使用者可利用NVIDIA的GeForce 8以後的GPU和較新的Quadro GPU進行計算。簡單來說CUDA就是GPU在解決大量的計算問題時的加速器。
3.cuDNN
cuDNN是NVIDIA為深度神經網路所打造的加速器,當你要用GPU來訓練模型時,我就會使用到它。
首先呢要先去Nvidia 先去看GPU的性能
因為Tensorflow會對GPU有所需求大致上可以分為
這裡的意思是如果你要安裝tensorflow 2.0以上的話你的Compute Capability要3.5以上
下表可以用你的GPU來對照Compute Capability
接下來就是安裝GPU driver
因為Tensorflow 會關係到你的cuDNN 和 CUDA 所以我會提供兩個表格給大家參考
(表一)
這是可以選擇的tensorflow版本 雖然現在應該已經到2.3了,但透過這張圖可以看到妳的CUDA的需求 接下來就要看你的gpu的本事有多大了
(表二)
所以如果你是WINDOWS版本要下載tensorflow2.0版本你的CUDA要求是10對照表二你的GPU driver 就要大於411.31
那我給大家看一下我下載的過程
一開始先去官網輸入你的電腦規格
接下來只要看你的版本有沒有大於等於你要下載Tensorflow需求就好了
到這裡只完成一半而已,(表一)還有一個重點就是python的版本,我一開始沒找這個圖時,我也一直卡住,明天再把剩下的呈現出來。
參考
https://medium.com/@johnnyliao/%E5%9C%A8win10%E4%B8%8A%E5%AE%89%E8%A3%9Dcuda-toolkit-cudnn-tensorflow-gpu-1-12%E4%BB%A5%E4%B8%8B%E5%8F%8A1-13%E4%BB%A5%E4%B8%8A-%E7%9A%84%E5%AE%89%E8%A3%9D%E7%B6%93%E9%A9%97%E5%88%86%E4%BA%AB-c792953b316f
https://zh.wikipedia.org/wiki/CUDA
https://kknews.cc/zh-tw/tech/zyjq3v3.html