作者a24285293 (雞眼)
看板DataScience
標題[問題] keras突然無法使用gpu訓練model
時間Tue Jul 24 22:22:17 2018
作業系統:(ex:mac,win10,win8,win7,linux,etc...)
WIN10
問題類別:(ex:ML,SVM,RL,DL,RNN,CNN,NLP,BD,Vis,etc...)
ML
使用工具:(ex:python,Java,C++,tensorflow,keras,MATLAB,etc...)
R keras tensorflow
問題內容:
各位先進好,小弟今天重啟Rstudio後使用keras訓練模型的時候發現變成cpu運算了,
https://imgur.com/XVzUXEI
於是我試著刪除conda環境後(conda env remove -n r-tensorflow)
重新install_keras(tensorflow = "gpu")
但是結果一樣是使用CPU運算,之前fit_generator的時候是能使用gpu運算的,
想請問發生了甚麼問題,有方法解決嗎?謝謝各位!
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 1.172.232.205
※ 文章網址: https://www.ptt.cc/bbs/DataScience/M.1532442142.A.DD0.html
※ 編輯: a24285293 (1.172.232.205), 07/24/2018 22:23:16
※ 編輯: a24285293 (1.172.232.205), 07/24/2018 22:43:58
→ atrix: 訊息裡有說,CUDA有問題 07/25 01:55
→ a24285293: 漏看了謝謝你 明天先試著從cuda版本檢查 07/25 02:11