THCudaCheck FAIL, RuntimeError: cublas runtime error

[danielkaifeng]

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : invalid argument
Traceback (most recent call last):
  File "test.py", line 255, in <module>
    example_dconv()
  File "test.py", line 175, in example_dconv
    output = dcn(input)
  File "/home/daniel/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/daniel/project/010_GAN/400_SR/Zooming-Slow-Mo-CVPR-2020/codes/models/modules/DCNv2/dcn_v2.py", line 107, in forward
    self.dilation, self.deformable_groups)
  File "/home/daniel/project/010_GAN/400_SR/Zooming-Slow-Mo-CVPR-2020/codes/models/modules/DCNv2/dcn_v2.py", line 27, in forward
    ctx.dilation[1], ctx.deformable_groups)
RuntimeError: cublas runtime error : the GPU program failed to execute at /pytorch/aten/src/THC/THCBlas.cu:425

[lyy-fighting]

Have you solve with it?