什么是＆＃34; GPU版本＆＃34; python中的numpy.fft包？

Question

我想在更短的时间内计算大信号（大样本大小）的fft（因此，GPU）。我尝试了以下模块：

• qsTr() - 不支持2D变换和2
的非幂
• PyFFT - 变换大小也不是任意的（权力为2,3,5）
• gpyfft - 仅返回N / 2 + 1个系数。会有问题 在进行复杂输入的转换时
• scikits.cuda - 任意输入，完全变换，支持2d 变换

什么是python包裹＆＃34; GPU版本＆＃34; numpy.fft包？ （可选：可以基于CUDA吗？）

Answer 1

如果您想要基于CUDA的库来计算FFT，其中变换大小是任意的，并支持1D，2D和3D FFT;那么你可能需要看看cuFFT。 http://docs.nvidia.com/cuda/cufft/#axzz410MycIFQ
注意：这是基于cuda的库，而不是python。