лучший способ провести FFT с помощью ускорения GPU с CUDA

Question

В python, что лучше всего запускать fft, используя вычисления cuda gpu? Я использую pyfftw для ускорения fftn, что примерно в 5 раз быстрее, чем numpy.fftn. Я хочу использовать pycuda для ускорения БПФ. Я знаю, что есть библиотека с именем pyculib, но мне всегда не удавалось установить ее, используя conda install pyculib. Есть какие-нибудь предложения?

Answer 1

Надеюсь, ответ еще не поздно, но мне также нужна была библиотека FFT, которая будет работать с CUDA без необходимости программировать ее самостоятельно.

Я использовал библиотеку PyFFT , которая, как мне кажется, устарела, но должна легко устанавливаться через Pip (например, pip install pyfft), которую я предпочитаю по сравнению с anaconda. Вы также можете попробовать Рейкна , что я считаю очень полезным, если вы также делаете Спектрограммы.

Надеюсь, это немного поможет! Я надеюсь, что talonmies был слишком отталкивающим:)

Answer 2

Я хочу использовать pycuda для ускорения БПФ

Ты не можешь. PyCUDA не имеет никакой встроенной поддержки FFT.