Getting this issue when I follow the dependency installation verbatim. Running the model on a NVIDIA L4 with nvcc --version 11.7.
This was solved by downloading torch with cu118:
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
I am concerned that 11.7 is deprecated for torch.fft usage.