TensorFlow GPU版安裝與使用

TensorFlow GPU版安裝與使用

深智數位資訊講堂 2021-08-26 13:32

深度學習 最佳入門邁向AI專題實戰 看更多請點我

本文探討在Windows作業環境下如何安裝TensorFlow,2.x版後安裝不再分CPU、GPU版本,直接執行下列指令,即可同時支援 CPU/GPU:

pip install tensorflow

執行時若出現『ImportError: DLL load failed: The specified module could not be found.』錯誤,就必須額外安裝MSVC 2019 runtime (https://aka.ms/vs/16/release/vc_redist.x64.exe),如果還有問題,可改用下列指令,安裝的版本可能舊一點:

conda install tensorflow

配備GPU可以在模型訓練時快上許多倍,目前TensorFlow只支援NVidia的顯示卡,要獲得NVidia GPU支援,除了驅動程式外,還要安裝以下工具箱:

  1. CUDA Toolkit
  2. cuDNN

相關工具的版本必須同時符合TensorFlow的版本、NVidia顯卡的限制,否則執行時會有錯誤發生,當TensorFlow升級時,版本也要隨之改變。建議顯卡不是GTX-1XXX或RTX等級以上,就不要安裝了,免得執行時常常發生記憶體不足的狀況,安裝時務必參考TensorFlow與NVidia官網的指引:

  1. https://www.tensorflow.org/install/gpu?hl=zh-tw
  2. https://developer.nvidia.com/cuda-toolkit-archive

安裝後的確認,可執行python,接著輸入import tensorflow,就可以看到是否有引用GPU函數庫。

另外,TensorFlow對GPU記憶體的垃圾回收(Garbage Collection)機制並不完美,因此,執行多個Jupyter Notebook常會出現GEMM錯誤:


表示GPU記憶體不足,因為Jupyter是一個網頁程式,即使關掉某一個Notebook檔案,網站仍然在執行中,並不代表該檔案的資源會被回收,通常要執行『Kernel > Restart』選單才會回收資源。另一個方式,就是限制GPU的使用配額,以下是TensorFlow 2.x版的方式,並不適用於1.x版:


也可選擇暫時不使用(Disable) GPU:

熱門文章
8.7超級大震恐襲台?專家大膽預警1事
8.7超級大震恐襲台?專家大膽預警1事

中天新聞

林莎泡湯剩毛巾遮體 卞慶華求加入「什麼都沒穿」
林莎泡湯剩毛巾遮體 卞慶華求加入「什麼都沒穿」

CTWANT

影/印度神童預言再中 最新預言警告台灣、美國、中國、印度小心10件事 特別注意5月
影/印度神童預言再中 最新預言警告台灣、美國、中國、印度小心10件事 特別注意5月

中天新聞

神秘失蹤!花蓮醫師賞鯨後莫名消失已5日 僅剩背包、名片丟船上
神秘失蹤!花蓮醫師賞鯨後莫名消失已5日 僅剩背包、名片丟船上

CTWANT

李芷婷無罩入鏡被讚深夜福利!網友直指:激凸了 她認不舒服
李芷婷無罩入鏡被讚深夜福利!網友直指:激凸了 她認不舒服

中天新聞

台餐飲集團「驚爆倒閉」!500人失業 30間門市一排倒
台餐飲集團「驚爆倒閉」!500人失業 30間門市一排倒

TVBS新聞網

倒數3個月!LINE「Keep」7/31終止服務 3招輕鬆備份
倒數3個月!LINE「Keep」7/31終止服務 3招輕鬆備份

中天新聞

印度神童最新預言曝光 台灣、美國、中國小心7件事 警告注意4、5、6月
印度神童最新預言曝光 台灣、美國、中國小心7件事 警告注意4、5、6月

中天新聞

立夏悄然而至!「4生肖」事業將飛黃騰達 成為眾人羨慕對象
立夏悄然而至!「4生肖」事業將飛黃騰達 成為眾人羨慕對象

中天新聞

再撐4天!5月「4生肖」財運大爆發 升官加薪買樂透也賺
再撐4天!5月「4生肖」財運大爆發 升官加薪買樂透也賺

TVBS新聞網

媽祖生日好運到!3生肖拜完「財產翻倍」 第1名正偏財旺炸
媽祖生日好運到!3生肖拜完「財產翻倍」 第1名正偏財旺炸

TVBS新聞網

88
0
分享