beyondguo/label_confusion_learning

关于代码无法成功调用gpu的问题

Opened this issue · 4 comments

想问一下源代码就是调用的cpu吗,我这边尝试了各种环境总是无法成功调用gpu

那应该不是代码的问题,你可以再检查一下环境配置,或者看看其他的keras程序能否调用gpu。keras应该在环境配置好之后自动调用gpu的

Kouuh commented

我想问下你能直接运行这个代码吗。我直接运行会出现找不到20NG这个文件。

想问一下源代码就是调用的cpu吗,我这边尝试了各种环境总是无法成功调用gpu

朋友我也是,检测gpu返回是true,但是跑的时候cpu都满了gpu也没有动静,请问你解决了吗

想问一下源代码就是调用的cpu吗,我这边尝试了各种环境总是无法成功调用gpu

就在刚刚我查看了一下cudnn的version,发现是8.2的版本,看来是我下载时搞错了。。。。。现在重现添加了7.6的版本,已经可以正常使用gpu了