In my experience, when you run into the cudnn error, the best way is going to pdb (not necessary) and convert the input, weight etc to CPU tensor, reexecute the command, you’ll find the error message from TH, which is more friendly.