Looks like cudnnGet* is returning the wrong algorithm. A workaround is to use torch.backends.cudnn.benchmark=True


  
      github.com/pytorch/pytorch
  
  
    



Issue: Handle bugs in cudnnGet* better 6


	
	opened by chenyuntc
	on 2017-05-05
	
	
	closed by chenyuntc
	on 
	


Simple code to reproduce the problem:
import torch
import torch.nn as nn
ec=nn.Conv3d(1,1,3,2,1).cuda()
input=torch.autograd.Variable(torch.ones(1,1,511,512,512),requires_grad=True).cuda()
ec(input).sum().backward()
it raises:
RuntimeError: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a...


 	bug
 	medium priority