Pytorch: 1x1 कर्नेल के साथ CUDA Conv2d करते समय त्रुटि।

1x1 कर्नेल के साथ Conv2d GPU पर काम नहीं कर रहा है, हालांकि यह CPU पर ठीक काम करता है:

net = nn.Conv2d(1, 6, kernel_size=(1,1))
net.cuda()
x = Variable(torch.randn(1, 1, 100, 100))
x.cuda()
net(x)

त्रुटि संदेश:

TypeError: FloatSpatialConvolutionMM_updateOutput received an invalid combination of arguments - got (int, torch.FloatTensor, torch.FloatTensor, torch.cuda.FloatTensor, torch.cuda.FloatTensor, torch.FloatTensor, torch.FloatTensor, int, int, int, int, int, int), but expected (int state, torch.FloatTensor input, torch.FloatTensor output, torch.FloatTensor weight, [torch.FloatTensor bias or None], torch.FloatTensor finput, torch.FloatTensor fgradInput, int kW, int kH, int dW, int dH, int padW, int padH)

मैंने torch.backends.cudnn.enabled = False साथ कुडन को अक्षम करने का प्रयास किया लेकिन फिर भी वही त्रुटि संदेश मिला।

मैं Ubuntu 14.04, Cuda 7.5, Cudnn 5.1.5, Python 3.5.2 का उपयोग करता हूं, और Pytorch बायनेरिज़ से स्थापित है।