search

robustsam / RobustSAM

RobustSAM: Segment Anything Robustly on Degraded Images (CVPR 2024 Highlight)
https://robustsam.github.io/
MIT License
152 stars 15 forks source link

RuntimeError: cuFFT error: CUFFT_INTERNAL_ERROR #8

Open SilenceGoo opened 1 month ago

SilenceGoo commented 1 month ago

Thansk for your great work. I followed commands step by step but got one error. Could guess what cause the problem? Thanks in advacne.

Xshell_60YH70OOPC
robustsam commented 1 month ago

I haven't encountered this problem before, but the components.py file was updated recently. Please try again to check whether the same problem occurs. Thank you.

SilenceGoo commented 1 month ago

Thanks for your reply. My setting environment may have problem as I updated the new code but error still. Will do more check.

robustsam commented 1 month ago

There is a similar discussion on https://github.com/pytorch/pytorch/issues/88038. I hope it helps.

SilenceGoo commented 1 month ago

Seems different CUDA verison or pytorch version might be the direct cause, will report here after more tests done.

SilenceGoo commented 1 month ago

Good news, I tried pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu120 instead of pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 to install bit lower version pkgs to match my CUDA:12.4, and it worked on RTX 4090.

Few warning info have no problems right?

MobaXterm_U8VQQi2GJ9