Pytorch: 1x1 커널로 CUDA Conv2d를 수행할 때 오류가 발생했습니다.

1x1 커널이 있는 Conv2d는 CPU에서는 잘 작동하지만 GPU에서는 작동하지 않습니다.

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 cudnn을 비활성화하려고 시도했지만 여전히 동일한 오류 메시지가 나타납니다.

Ubuntu 14.04, Cuda 7.5, Cudnn 5.1.5, Python 3.5.2를 사용하고 Pytorch는 바이너리에서 설치됩니다.