iT邦幫忙

2

Spyder成功引入tensorflow但沒用到gpu 資源

radd 2022-09-02 21:14:122486 瀏覽
  • 分享至 

  • xImage

第一次發問若有不對的地方還請多指教
先前有按照網路上教學安裝了nvidia驅動、CUDA、Cudnn
並有設置好anaconda環境 打開spyder可以成功引入tensor flow 也可以顯示版本
但在跑程式碼時發現都是記憶體和cpu被使用 gpu 幾乎沒用到
想請問這樣的狀況要怎麼解決

增廣建文 iT邦研究生 5 級 ‧ 2022-09-03 12:17:34 檢舉
估計是cuda + cudnn的版本沒跟TF版本對上
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

2 個回答

0
haward79
iT邦研究生 1 級 ‧ 2022-09-03 03:52:52

我是建議你把詳細安裝過程放上來
或是把你看的教學文章放上來
另外,你跑的程式也需要放上來
這樣比較容易找出問題
不然網友們估計只能通靈或瞎猜了!

以下就舉例幾種可能的原因吧!

tensorflow 裝錯

請安裝 tensorflow-gpu,若是安裝 tensorflow 是沒用的。

軟體版本不對

你的 CUDA、CUDNN、tensorflow-gpu 的版本要對的起來,不然即使裝好了,執行程式時也會跑出錯誤、無法正常使用。另外有時候會需要另外設置環境變數等參數,例如:同時安裝多個CUDA版本或是舊的CUDNN未被解除安裝之類的。
版本對照請參考 官方說明

虛擬環境沒進對

你的相關套件可能安裝在指定的虛擬環境下,但是你執行程式時沒有進到那個虛擬環境中。

程式本來就用不到GPU

你跑的程式本來就無法使用GPU去跑或者撰寫的時候不是為GPU運算而寫的,所以即使相關軟硬體都正常,GPU也不會參與運算,還是CPU要背運算的責任。

增廣建文 iT邦研究生 5 級 ‧ 2022-09-03 12:07:13 檢舉

請安裝 tensorflow-gpu,若是安裝 tensorflow 是沒用的。

TF > 2.0後不用特別裝GPU版喔!

zzssc061u iT邦新手 5 級 ‧ 2022-10-23 12:48:15 檢舉

不安裝tensorflow-gpu,tf.config.experimental.list_physical_devices('GPU')都沒東西,是正常?

0
WenTingTseng
iT邦新手 4 級 ‧ 2022-09-03 11:10:10

沒有安裝到tensor flow GPU的版本吧?!

我要發表回答

立即登入回答