PyCuda mem_alloc 初始化错误

Question

in desaturate_image
    redarray_gpu = cuda.mem_alloc(self.redarray.nbytes)
pycuda._driver.LogicError: cuMemAlloc failed: initialization error

我在这一行收到上述错误：

redarray_gpu = cuda.mem_alloc(self.redarray.nbytes)

可能是什么原因？

Answer 1

import pycuda.autoinit

此导入语句修复了该问题。我最好的猜测是设备没有初始化