CUFFT_INTERNAL_ERROR when running HorNet

wangzy22 / P2P

[NeurIPS 2022 Spotlight] P2P: Tuning Pre-trained Image Models for Point Cloud Analysis with Point-to-Pixel Prompting

https://p2p.ivg-research.xyz

MIT License

119 stars 9 forks source link

Closed detectRecog closed 1 year ago

detectRecog commented 1 year ago

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

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

wangzy22 commented 1 year ago

Hi, thanks for your interest in our work!

We didn't encounter such error about HorNet under our env. What's your pytorch version and nvcc version?