PyCuda mem_alloc 初始化错误

PyCuda mem_alloc initialization error

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)

可能是什么原因?

import pycuda.autoinit

这个导入语句解决了这个问题。我最好的猜测是设备没有初始化