PSPNet icon indicating copy to clipboard operation
PSPNet copied to clipboard

Running eval_all.m , use cpu can get correct result, but GPU got a wrong result

Open Hutai opened this issue 8 years ago • 2 comments

Why I use cpu to run the eval_all.m got correct pictures of prediction, but using gpu can not get what I want that it output whole black or whole yellow images?

Hutai avatar Mar 10 '17 13:03 Hutai

I have solved this problem, it is cudnn version of the problem, change to other versions of cudnn or uncomment cudnn configuration information

Hutai avatar Mar 13 '17 06:03 Hutai

safest bet might just be to run without cuDNN if this is the case. only accounts for about a 40% speedup on eval and a 35% speedup on training

balloch avatar Mar 17 '17 07:03 balloch