在python中使用cuda计算pi的近似值

Question

我想用这个简单的公式来计算pi的近似值： pi approx formula

这是我的代码：

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

使用此PyCUDA允许您在GPU上运行代码，这里是docs

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

另一个替代方案是PyGPU

Answer 2

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