Error when running pytorch1.9 version about dcnv2 HOT 2 CLOSED

(rgo-py37) fy@neu:~/DCNv2-pytorch_1.7/test$ python test.py
error in modulated_deformable_im2col_cuda: no kernel image is available for execution on the device
Traceback (most recent call last):
File "test.py", line 284, in
example_dconv()
File "test.py", line 202, in example_dconv
error.backward()
File "/home/fy/anaconda3/envs/rgo-py37/lib/python3.7/site-packages/torch/tensor.py", line 221, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/home/fy/anaconda3/envs/rgo-py37/lib/python3.7/site-packages/torch/autograd/init.py", line 132, in backward
allow_unreachable=True) # allow_unreachable flag
File "/home/fy/anaconda3/envs/rgo-py37/lib/python3.7/site-packages/torch/autograd/function.py", line 89, in apply
return self._forward_cls.backward(self, *args) # type: ignore
File "/home/fy/anaconda3/envs/rgo-py37/lib/python3.7/site-packages/torch/autograd/function.py", line 210, in wrapper
outputs = fn(ctx, *args)
File "/home/fy/DCNv2-pytorch_1.7/dcn_v2.py", line 61, in backward
ctx.deformable_groups,
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

When I run the pytorch 1.7 version of the test.py, it also raise an error.

from dcnv2.

The two environments I installed above can train the model, but neither of them can use DCN.

from dcnv2.