使用GPU跑python程序

需要用到numba + cuda 。 cuda的安装直接找网上教程。numba的安装，如果安装有anaconda可以在command prompt 输入：

conda install numba
conda install cudatoolkit 

cudatoolkit安装有了，可以不用在安装。

下面是运用的例子

from numba import jit, cuda  # 导入numba

@jit(target_backend='cuda')  # 在要用GPU运算函数前加上这一句
def caculate():
    b = 0
    for i in range(1024**3):
        b = b + i
    return b
print(caculate())