gpt4 book ai didi

python - PyCuda mem_alloc 初始化错误

转载 作者:行者123 更新时间:2023-12-05 08:54:12 25 4
gpt4 key购买 nike

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

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

关于python - PyCuda mem_alloc 初始化错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51030316/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com