使用 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

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

免责声明：我在 ArrayFire 工作。

Answer 2

有PyFFT，即open-source，基于Apple 的implementation。免责声明：我在 PyFFT 上工作 :)