运行 python cuda 程序出错

Question

我是 python 内核编程的新手，为了学习，我遵循了这个 link。在尝试运行示例 Cuda python 程序时，出现如下错误。我不知道，这是关于什么的？请帮助我解决这个问题，以便我继续学习。

Traceback (most recent call last):
File "numpycuda.py", line 17, in <module>
my_kernel[blockspergrid, threadsperblock](data)

File "/home/face/.local/lib/python2.7/site- 
packages/numba/cuda/simulator/kernel.py", line 103, in 
__getitem__
normalize_kernel_dimensions(*configuration[:2])

File "/home/face/.local/lib/python2.7/site- 
packages/numba/cuda/errors.py", line 38, in 
normalize_kernel_dimensions
griddim = check_dim(griddim, 'griddim')

File "/home/face/.local/lib/python2.7/site- 
packages/numba/cuda/errors.py", line 33, in check_dim
% (name, dim)).

TypeError: griddim must be a sequence of integers, got [1.0]

Python 节目

from __future__ import division
from numba import cuda
import numpy
import math

# CUDA kernel
@cuda.jit
def my_kernel(io_array):
   pos = cuda.grid(1)
   if pos < io_array.size:
      io_array[pos] *= 2 # do the computation

# Host code   
data = numpy.ones(256)
threadsperblock = 256
blockspergrid = math.ceil(data.shape[0] / threadsperblock)
my_kernel[blockspergrid, threadsperblock](data)
print(data)

我安装了 numba、CUDA 和 numpy 库，可能是什么问题？我正在使用 python 版本 2.7.12

Answer 1

Numba 内核启动要求执行参数为整数或整数元组。您对 math.ceil(data.shape[0] / threadsperblock) 的使用产生了一个浮点数，将其用作执行参数是非法的。

你可以这样做：

data = numpy.ones(250)
threadsperblock = 64
blockspergrid = (data.shape[0] + threadsperblock - 1) // threadsperblock
my_kernel[blockspergrid, threadsperblock](data)

哪个应该可以正常工作

运行 python cuda 程序出错

Error while running the python cuda program

python

cuda

python-2.7

numba