CUBLAS_ STATUS_ ALLOC_ FAILED

resolvent:

Extract binding CUDA

torch.rand (1, 3, 10, 10).cuda(7)

It’s OK in training, but not in prediction.

>>> input = torch.rand(1, 3, 10, 10)
>>> kh, kw = 3, 3
>>> weight = torch.rand(5, 3, kh, kw)
>>> # offset should have the same spatial size as the output
>>> # of the convolution. In this case, for an input of 10, stride of 1
>>> # and kernel size of 3, without padding, the output size is 8
>>> offset = torch.rand(5, 2 * kh * kw, 8, 8)
>>> out = deform_conv2d(input, offset, weight)