安裝Keras, CUDA, cuDNN之後無法使用GPU進行運算?
2023/09/16 下午 06:56
Keras 安裝與介紹
A. Hung
觀看數:171
回答數:1
收藏數:0
想請問,目前按照課程內容安裝Keras, CUDA, cuDNN, 目前裝置是筆記型電腦,搭配RTX 4050顯卡。 但透過 tensorflow.config.list_physical_devices('GPU') 沒有檢查到可以使用的GPU, 執行Keras程式碼來訓練類神經網路時,在「工作管理員」的「效能」之中也沒有看到GPU被使用。 我該透過什麼方式在訓練過程中使用GPU呢? 或是要檢查哪個部分或哪個程式的版本?
回答列表
-
2023/09/18 上午 11:32王健安贊同數:0不贊同數:0留言數:1
A. Hung 您好, GPU、cuda、cudnn 三者有相互對應的版本, 安裝順序上也得先裝 cuda,再裝 cudnn, 因為近兩年 tensorflow 已經把 gpu 版融合至一般版本中, 已經沒有額外的 tensorflow-gpu 模組, 因此我猜測是安裝的某個環節出問題, 我有找到去年有其他人寫的參考資料,供您參考, 也建議您找資源時要注意避免找到太舊的資源。 https://unimimic.github.io/posts/install-cuda-cudnn-win10/