时间戳

首页 > 解决方案 > PyCuda mem_alloc 初始化错误

python - PyCuda mem_alloc 初始化错误

问题描述

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)

可能是什么原因？

标签： pythonarraysnumpypycharmpycuda

解决方案

import pycuda.autoinit

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

推荐阅读