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:

熱門文章
詭異!寶林茶室「只有3/20沒人中毒」 專家曝2可能原因
詭異!寶林茶室「只有3/20沒人中毒」 專家曝2可能原因

中天新聞

中毒案奪2命!他揪公司餐驚「寶林員工全逃過」 餐飲業內幕曝
中毒案奪2命!他揪公司餐驚「寶林員工全逃過」 餐飲業內幕曝

TVBS新聞網

郵局發錢了!學生、家長嗨領「最高3千元」 錯過捶心肝
郵局發錢了!學生、家長嗨領「最高3千元」 錯過捶心肝

TVBS新聞網

新竹冰沙「陷食安風波」?多人發燒嘔吐 衛生局派員稽查
新竹冰沙「陷食安風波」?多人發燒嘔吐 衛生局派員稽查

TVBS新聞網

「寶林茶室」出事粿條黎仿軒也有吃! 爸媽還打包回家
「寶林茶室」出事粿條黎仿軒也有吃! 爸媽還打包回家

中天新聞

女警想請假遭疑裝病「檢查出癌症四期」 所長刁難才給假…下場出爐
女警想請假遭疑裝病「檢查出癌症四期」 所長刁難才給假…下場出爐

CTWANT

直擊/中市焢肉飯食材露天放  網「Google街景都有  業者似乎習以為常」
直擊/中市焢肉飯食材露天放  網「Google街景都有  業者似乎習以為常」

記者爆料網

台北市「寶林茶室案」4重症病例持續惡化 衛生局:1病例肝腎衰竭、腦部病變
台北市「寶林茶室案」4重症病例持續惡化 衛生局:1病例肝腎衰竭、腦部病變

中天新聞

鴻海、廣達...黃仁勳欽點「8家將」護體輝達:AI是新的工業革命「台灣處於最中心位置」
鴻海、廣達...黃仁勳欽點「8家將」護體輝達:AI是新的工業革命「台灣處於最中心位置」

今周刊

好市多新品曝光!「超頂麵包捲」出爐 會員嗨翻:終於等到了
好市多新品曝光!「超頂麵包捲」出爐 會員嗨翻:終於等到了

TVBS新聞網

海鮮公司變電箱下挖「密道」偷電 電費大減437萬度竊得2702萬元
海鮮公司變電箱下挖「密道」偷電 電費大減437萬度竊得2702萬元

CTWANT

快訊/「寶林茶室」禍首曝光!衛福部專家會議推測「這毒素」或成元兇
快訊/「寶林茶室」禍首曝光!衛福部專家會議推測「這毒素」或成元兇

中天新聞

獨家/汪小菲求婚成功!示愛新歡Mandy:我的老婆來自台北
獨家/汪小菲求婚成功!示愛新歡Mandy:我的老婆來自台北

CTWANT

北台灣變天!「雷雨胞將襲」影響程度曝 下周暖熱如夏
北台灣變天!「雷雨胞將襲」影響程度曝 下周暖熱如夏

中天新聞

水逆凶狠殺到!5星座「災厄纏身」受重擊 牡羊最慘倒大楣了
水逆凶狠殺到!5星座「災厄纏身」受重擊 牡羊最慘倒大楣了

TVBS新聞網

他時常蕁麻疹發作,竟是「1提神飲品」惹禍! 不少人早上都要喝
他時常蕁麻疹發作,竟是「1提神飲品」惹禍! 不少人早上都要喝

健康醫療網

離開科技業跳槽傳產!頂大生反悔「還能回鍋嗎」 過來人狂勸退
離開科技業跳槽傳產!頂大生反悔「還能回鍋嗎」 過來人狂勸退

TVBS新聞網

寶林茶室食物中毒案重大發現!死者解剖確認驗出「米酵菌酸」!
寶林茶室食物中毒案重大發現!死者解剖確認驗出「米酵菌酸」!

食力foodNEXT

寶林2死!台大醫喊避免「這種飯、麵」重複加熱 米酵菌酸更猛爆
寶林2死!台大醫喊避免「這種飯、麵」重複加熱 米酵菌酸更猛爆

TVBS新聞網

汪小菲喜曝求婚成功...命理師驚見女方「命格特殊」 難怪罩得住失控哥
汪小菲喜曝求婚成功...命理師驚見女方「命格特殊」 難怪罩得住失控哥

CTWANT

88
0
分享