CUBLAS_STATUS_ALLOC_FAILED
解决方法:
提取绑定cuda
torch.rand(1, 3, 10, 10).cuda(7)
在训练时ok,只预测时结果不对。
>>> 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)