2

In python, what is the best to run fft using cuda gpu computation? I am using pyfftw to accelerate the fftn, which is about 5x faster than numpy.fftn. I want to use pycuda to accelerate the fft. I know there is a library called pyculib, but I always failed to install it using conda install pyculib. Is there any suggestions?

billinair
  • 93
  • 1
  • 11

2 Answers2

1

I want to use pycuda to accelerate the fft

You can't. PyCUDA has no built-in FFT support of any kind.

talonmies
  • 70,661
  • 34
  • 192
  • 269
1

Hopefully this isn't too late of answer, but I also needed a FFT Library that worked will with CUDA without having to programme it myself.

I was using the PyFFT Library which I think is deprecated but should be able to be easily installed via Pip (e.g. pip install pyfft) which I much prefer over anaconda. You could also try Reikna, which I have found very useful if you are doing Spectrograms as well.

Hope this helps a bit! I hope talonmies was too off-putting :)

Clement
  • 367
  • 2
  • 18