在执行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 这行是灰色的,给人的感觉是没有用,但是如果没有这句话就会报上面的错误。