在 Python 中使用 CUDA 计算 pi 的近似值

Question

我想用这个简单的公式来计算pi的近似值：圆周率近似公式

这是我的代码：

def pi(iteration):
    pi = 0.0
    sign = 1
    b = 1.0
    for i in range(iteration):
        pi += 4.0*sign/b
        b += 2
        sign *= -1
    print pi
pi(100000)

如何使此代码由 GPU 计算？

Answer 1

你可以使用 Numba。这是一个专门针对 CUDA 使用 python 计算 pi 值的好例子。https://github.com/numba/numba/blob/master/examples/pi.py

Answer 2

使用这个PyCUDA允许你在GPU上运行代码，这里有文档

但我必须说，只是弄乱它不是一个好主意。小心过热。

还有另一种选择是这个 PyGPU