本博客参照学习文档https://documen.tician.de/pycuda/array.html
通过GPUArray 矩阵类可以将numpy中的数组和矩阵直接转换成cuda可处理类型,该步骤即将CPU中的数据复制到GPU中。
导入gpuArray和numpy
import pycuda.gpuarray as gpuarray
import numpy as np
将numpy ndarray转成gpuArray,如下:
gpuarray.to_gpu(ary)
这里需要主要的是,由于gpu最支持的数据类型是单精度浮点类型,因此ary最好定义为np.float32类型,如:
gpu_ary = gpuarray.to_gpu(np.random.rand(m,n).astype(np.float32))
类似的,可以采用一下方式进行gpuArray创建
gpu_ary = gpuarra