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槍 逃逸毒品通緝犯今晨85大樓落網
遭衝撞警連開8槍 逃逸毒品通緝犯今晨85大樓落網

CTWANT

威力彩得主遭詐2/黑心宮主餵尿治信眾 起乩誆稱太上老君要錢要名車
威力彩得主遭詐2/黑心宮主餵尿治信眾 起乩誆稱太上老君要錢要名車

CTWANT

趁國小女兒同學夜宿他家 狼父伸魔爪猥褻下場曝光
趁國小女兒同學夜宿他家 狼父伸魔爪猥褻下場曝光

CTWANT

全國壓軸一「2022高雄愛.月熱氣球」再不搶等明年
全國壓軸一「2022高雄愛.月熱氣球」再不搶等明年

今傳媒

國3和美交流道奇景! 大貨車暗藏玄機「夾手車輪」
國3和美交流道奇景! 大貨車暗藏玄機「夾手車輪」

記者爆料網

運動反而害你背痛腰痛、膝蓋痛?其實是核心肌群出問題!印度瑜珈冠軍教你「5動作」舒緩
運動反而害你背痛腰痛、膝蓋痛?其實是核心肌群出問題!印度瑜珈冠軍教你「5動作」舒緩

良醫健康網

24歲男跳樓逃回台 曝柬國最年輕奴工僅15歲
24歲男跳樓逃回台 曝柬國最年輕奴工僅15歲

CTWANT

新北無照少年偷騎機車放風出車禍 老翁遭撞飛送醫不治
新北無照少年偷騎機車放風出車禍 老翁遭撞飛送醫不治

CTWANT

中元節供品久放恐滋生細菌!  營養師授5大保存技巧
中元節供品久放恐滋生細菌! 營養師授5大保存技巧

潮健康

可不可熟成紅茶2飲品「腸桿菌超標」 複驗未過挨罰3萬
可不可熟成紅茶2飲品「腸桿菌超標」 複驗未過挨罰3萬

TVBS新聞網

南投移工仲介遭「鐵絲綁雙手」死狀慘 高大成見凶器推測:兇手至少2人
南投移工仲介遭「鐵絲綁雙手」死狀慘 高大成見凶器推測:兇手至少2人

CTWANT

骨頭不好不能運動?骨科醫師破解迷思,親授「運動治療法」4大重點,每週3次比只吃消炎藥還有用!
骨頭不好不能運動?骨科醫師破解迷思,親授「運動治療法」4大重點,每週3次比只吃消炎藥還有用!

Bella.tw儂儂

「木蘭」颱風不來!這區降雨範圍擴大 嚴防午後大雨
「木蘭」颱風不來!這區降雨範圍擴大 嚴防午後大雨

TVBS新聞網

威力彩得主遭詐1/中彩券成億萬富翁 皮件老董捐3.7億蓋廟遭侵吞
威力彩得主遭詐1/中彩券成億萬富翁 皮件老董捐3.7億蓋廟遭侵吞

CTWANT

搶到票恐沒車搭 台鐵工會:中秋100%全停駛
搶到票恐沒車搭 台鐵工會:中秋100%全停駛

TVBS新聞網

打造海線新地標 「台中海洋館」預計112年底開幕營運
打造海線新地標 「台中海洋館」預計112年底開幕營運

民生頭條

刷完牙不要漱口!牙醫揭6大錯誤觀念:牙膏用太多也NG
刷完牙不要漱口!牙醫揭6大錯誤觀念:牙膏用太多也NG

TVBS新聞網

中央國旅補助台中加碼 每人可享2300元補貼
中央國旅補助台中加碼 每人可享2300元補貼

中廣新聞

頂新劣油官司纏訟5年 統一求償10億最後1.7億元和解
頂新劣油官司纏訟5年 統一求償10億最後1.7億元和解

上報Up Media

民眾淨灘拾獲異香怪石!高大:台灣本島海灘發現龍涎香真品首例
民眾淨灘拾獲異香怪石!高大:台灣本島海灘發現龍涎香真品首例

焦點時報

1歲嬰連吐3天、不願進食   竟是「誤食這小物品」害到切腸子
1歲嬰連吐3天、不願進食 竟是「誤食這小物品」害到切腸子

匯流新聞網CNEWS

更年期常常下背痛、關節痛?3大保健秘訣一次看
更年期常常下背痛、關節痛?3大保健秘訣一次看

健康醫療網

蕭淑慎服刑洗澡「胸部被看光光」:讓我摸一下? 上廁所慘遭霸凌⋯腳全伸出來
蕭淑慎服刑洗澡「胸部被看光光」:讓我摸一下? 上廁所慘遭霸凌⋯腳全伸出來

CTWANT

5大星座迎好運!塔羅牌老師點名「獅子座」全力拚事業…升職加薪不失望
5大星座迎好運!塔羅牌老師點名「獅子座」全力拚事業…升職加薪不失望

藝點新聞

通緝女在逃多年 終遭岡山警緝獲
通緝女在逃多年 終遭岡山警緝獲

【『好報』報系:台灣好報】

不只柬埔寨!全球反詐騙組織點名「這5國家」:不要去就對了
不只柬埔寨!全球反詐騙組織點名「這5國家」:不要去就對了

民視新聞

立秋來了!命理師曝4生肖「5運神炸」 錢狂賺擋不住了
立秋來了!命理師曝4生肖「5運神炸」 錢狂賺擋不住了

TVBS新聞網

中年全職煮婦罹肺癌 恐與「家中1物」有關
中年全職煮婦罹肺癌 恐與「家中1物」有關

健康醫療網

大部分腦瘤是後天造成的!腦神經權威教你這樣做「預防腦瘤」
大部分腦瘤是後天造成的!腦神經權威教你這樣做「預防腦瘤」

良醫健康網

全國首創!熱氣球與300架無人機共演  饒慶鈴:耳目一新感受
全國首創!熱氣球與300架無人機共演 饒慶鈴:耳目一新感受

News586

88
0
分享