作者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://webptt.com/cn.aspx?n=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
1F:→ atrix: 讯息里有说,CUDA有问题 07/25 01:55
2F:→ a24285293: 漏看了谢谢你 明天先试着从cuda版本检查 07/25 02:11