RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:425

[GilbertTam]

When i use cuda version it face some error below,

My environment: torch 1.1.0
Driver Version: 440.26 CUDA Version: 10.2 ─╯

python testcuda.py ─╯ THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : invalid argument Traceback (most recent call last): File "testcuda.py", line 255, in example_dconv() File "testcuda.py", line 175, in example_dconv output = dcn(input) File "/home/infor/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in call result = self.forward(*input, **kwargs) File "/home/infor/DCNv2/dcn_v2.py", line 128, in forward self.deformable_groups) File "/home/infor/DCNv2/dcn_v2.py", line 31, in forward ctx.deformable_groups) RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:425

Can help me solve this problem?

[Haoju-Leng]

Hi I have faced the same issue right now. Do you know how to solve it now? Thanks!