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:

熱門文章
花蓮鐵皮屋惡火 一家六口4人罹難
花蓮鐵皮屋惡火 一家六口4人罹難

CTWANT

桃園7/16送煞儀式長達8小時!路線曝光 居民傻眼
桃園7/16送煞儀式長達8小時!路線曝光 居民傻眼

中天新聞

曬太陽是天然的血壓藥?研究曝每天「曬太陽30分鐘」防心血管疾病
曬太陽是天然的血壓藥?研究曝每天「曬太陽30分鐘」防心血管疾病

uho優活健康網

花蓮鐵皮屋惡火變「鐵棺材」…一家6口破碎 4人成焦屍
花蓮鐵皮屋惡火變「鐵棺材」…一家6口破碎 4人成焦屍

中天新聞

洋蔥是超級護腎食物 醫生親解:「這樣吃」才對
洋蔥是超級護腎食物 醫生親解:「這樣吃」才對

健康醫療網

新颱風恐襲台!專家警告:週末恐直撲台灣
新颱風恐襲台!專家警告:週末恐直撲台灣

品觀點傳媒

鄧佳華自曝「交女友、月入6萬」全是膨風? 前老闆公布真實數字開罵
鄧佳華自曝「交女友、月入6萬」全是膨風? 前老闆公布真實數字開罵

CTWANT

卓榮泰定調「不普發1萬元」? 網友怒了
卓榮泰定調「不普發1萬元」? 網友怒了

中廣新聞

快訊/立法院三讀通過!博愛座改名「優先席」
快訊/立法院三讀通過!博愛座改名「優先席」

中天新聞

知名親子樂園「最後2館」9月關 曾擁10家分館全倒
知名親子樂園「最後2館」9月關 曾擁10家分館全倒

TVBS新聞網

颱風「薇帕」最快24小時將生成 最新路徑、發布警戒機率曝光
颱風「薇帕」最快24小時將生成 最新路徑、發布警戒機率曝光

上報Up Media

疲老闆離婚撕家庭醜陋面 答應結紮「妻卻出軌3男」:和別人生孩子
疲老闆離婚撕家庭醜陋面 答應結紮「妻卻出軌3男」:和別人生孩子

CTWANT

颱風恐撲台!路徑估「貫穿台灣」 專家曝新預測
颱風恐撲台!路徑估「貫穿台灣」 專家曝新預測

TVBS新聞網

快訊/小哥哥艾理捲假壯陽藥風波 發聲明切割稱受害者
快訊/小哥哥艾理捲假壯陽藥風波 發聲明切割稱受害者

上報Up Media

台中西屯、北區汰換管線! 11里7/15起「分區停水3天」
台中西屯、北區汰換管線! 11里7/15起「分區停水3天」

中天新聞

台中18歲女當街被刀割頸! 警方鎖定前男友追查中
台中18歲女當街被刀割頸! 警方鎖定前男友追查中

CTWANT

記憶力變差是老化還是失智症前兆?醫:「這些狀況」可能是失智症
記憶力變差是老化還是失智症前兆?醫:「這些狀況」可能是失智症

健康醫療網

88
0
分享