使用cuda与GPU加速进行fft的最佳方法

时间:2018-06-27 15:12:08

标签: python cuda cufft

在python中,使用cuda gpu计算运行fft的最佳方法是什么? 我正在使用pyfftw来加速fftn,它比numpy.fftn快5倍。 我想使用pycuda来加快速度。我知道有一个名为pyculib的库,但是我始终无法使用conda install pyculib安装它。 有什么建议吗?

2 个答案:

答案 0 :(得分:1)

  

我想用pycuda来加速fft

不能。 PyCUDA没有任何内置的FFT支持。

答案 1 :(得分:1)

希望这还不算太晚,但是我还需要一个可以与CUDA一起使用的FFT库,而不必自己编程。

我使用的是PyFFT Library,我认为它已被弃用,但应该可以通过Pip(例如pip install pyfft)轻松安装,相对于anaconda,我更喜欢它。您也可以尝试Reikna,如果您也执行频谱图,我发现它非常有用。

希望这会有所帮助!我希望绒布太讨厌了:)