pycuda._driver.LogicError: explicit_context_dependent failed: invalid device context - no currently
在执行tensorRT推理的时候出现了错误,详细信息如下:D:\cv\ConNext_demo\testtrt.py:32: DeprecationWarning: Use set_optimization_profile_async instead.context.active_optimization_profile = 0Traceback (most recent call last):F
·
在执行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 这行是灰色的,给人的感觉是没有用,但是如果没有这句话就会报上面的错误。
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献2条内容
所有评论(0)