在python中,使用cuda gpu计算运行fft的最佳方法是什么?
我正在使用pyfftw来加速fftn
,它比numpy.fftn
快5倍。
我想使用pycuda
来加快速度。我知道有一个名为pyculib
的库,但是我始终无法使用conda install pyculib
安装它。
有什么建议吗?
答案 0 :(得分:1)
我想用pycuda来加速fft
不能。 PyCUDA没有任何内置的FFT支持。
答案 1 :(得分:1)
希望这还不算太晚,但是我还需要一个可以与CUDA一起使用的FFT库,而不必自己编程。
我使用的是PyFFT Library,我认为它已被弃用,但应该可以通过Pip(例如pip install pyfft
)轻松安装,相对于anaconda,我更喜欢它。您也可以尝试Reikna,如果您也执行频谱图,我发现它非常有用。
希望这会有所帮助!我希望绒布太讨厌了:)