python调用cuda程序_使用Python写CUDA程序的方法

使用Python写CUDA程序的方法

来源：中文源码网    浏览： 次    日期：2018年9月2日

【下载文档:  使用Python写CUDA程序的方法.txt 】

(友情提示:右键点上行txt文档名->目标另存为)

使用Python写CUDA程序的方法 使用Python写CUDA程序有两种方式：

* Numba

* PyCUDA

numbapro现在已经不推荐使用了，功能被拆分并分别被集成到accelerate和Numba了。

例子

numbaNumba通过及时编译机制(JIT)优化Python代码，Numba可以针对本机的硬件环境进行优化，同时支持CPU和GPU的优化，并且可以和Numpy集成，使Python代码可以在GPU上运行，只需在函数上方加上相关的指令标记，

如下所示：

import numpy as np

from timeit import default_timer as timer

from numba import vectorize@vectorize(["float32(float32, float32)"], target='cuda')

def vectorAdd(a, b):

return a + bdef main():

N = 320000000 A = np.ones(N, dtype=np.float32 )

B = np.ones(N, dtype=np.float32 )

C = np.zeros(N, dtype=np.float32 ) star