我一直在寻找一些简单的方法来编写FFT以在我的显卡上工作(这是NVIDIA最近推出的一款支持CUDA 3.0的显卡)。我目前的选择是学习C语言,然后学习CUDA的特殊C版本,或者使用一些python CUDA函数。我不想学习C语言,因为我只用高级语言编程。我查看了pyCUDA和其他在python中使用显卡的方法,但我找不到任何只能与python代码一起使用的快速傅立叶变换库。一些库/项目似乎可以处理类似的项目(CUDAmat,Theano),但遗憾的是我没有找到FFT。
有没有一个函数可以用我的显卡和numpy.fft.fft2()
做同样的事情?
编辑:开源解决方案的加分。
发布于 2012-09-15 22:17:33
还有PyFFT,它是open-source,基于苹果的(有些有限的) implementation。免责声明:我在PyFFT上工作:)
发布于 2012-08-25 07:33:44
是的,ArrayFire有一个2-D FFT for Python。
免责声明:我在ArrayFire上工作。
https://stackoverflow.com/questions/12117327
复制相似问题