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

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐