CUFFT_INTERNAL_ERROR when running HorNet

[detectRecog]

{RuntimeError}cuFFT error: CUFFT_INTERNAL_ERROR at "HorNet.py", line 64, in forward x2 = torch.fft.rfft2(x2, dim=(2, 3), norm='ortho')

Hello, Have you ever come across this error about HorNet?

[raoyongming]

Hi, thanks for your interest in our work.

We didn't come across the error in our experiments. The error seems to be related to the version of Pytorch or GPU driver.

According to the CUDA document, CUFFT_INTERNAL_ERROR indicates an internal driver error was detected. I think updating CUDA or Pytorch might solve this problem.