使用CUDA或OpenCL在python中进行多维FFT

Question

我一直在寻找简单的方法来编程FFT以在我的显卡上工作（这是最近支持CUDA 3.s​​omething的NVIDIA）。 我目前的选择是学习C，然后学习CUDA的特殊C版本，或使用一些python CUDA函数。我还不想学习C，因为我只用高级语言编程。 我查看了pyCUDA以及在python中使用我的图形卡的其他方法，但我找不到任何只能与python代码一起使用的FFT库。 一些图书馆/项目似乎解决了类似的项目（CUDAmat，Theano），但遗憾的是我没有发现FFT。

是否存在可以使用我的图形卡与numpy.fft.fft2()执行相同操作的功能？

编辑：开源解决方案的奖励点。

Answer 1

有PyFFT，open-source，基于Apple（有限）implementation。免责声明：我在PyFFT工作：）

Answer 2

是的，ArrayFire有一个2-D FFT for Python。

免责声明：我在使用ArrayFire。