pycuda._driver.LogicError: explicit_context_dependent failed: invalid device context - no currently

Posted 2022-03-12 AI浩

在执行tensorRT推理的时候出现了错误，详细信息如下：

D:\\cv\\ConNext_demo\\testtrt.py:32: DeprecationWarning: Use set_optimization_profile_async instead.
  context.active_optimization_profile = 0
Traceback (most recent call last):
  File "D:\\cv\\ConNext_demo\\testtrt.py", line 37, in <module>
    d_input = cuda.mem_alloc(image.nbytes)  # 分配输入的内存。
pycuda._driver.LogicError: explicit_context_dependent failed: invalid device context - no currently active context?

这是pycuda没有初始化，执行初始化即可：

import pycuda.driver as cuda
import pycuda.autoinit

import pycuda.autoinit 这行是灰色的，给人的感觉是没有用，但是如果没有这句话就会报上面的错误。