CUFFT_INTERNAL_ERROR on RTX 4090 #88038
Description
🐛 Describe the bug
>>> import torch
>>> torch.fft.rfft(torch.randn(1000).cuda())
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
RuntimeError: cuFFT error: CUFFT_INTERNAL_ERROR
There is a discussion on https://forums.developer.nvidia.com/t/bug-ubuntu-on-wsl2-rtx4090-related-cufft-runtime-error/230883/7 .
Versions
Using pytorch installed with
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
cc @ezyang @gchanan @zou3519 @peterjc123 @mszhanyi @skyline75489 @nbcsm @ngimel @mruberry @peterbell10