TensorRT 多线程使用报错

最新推荐文章于 2025-07-19 10:02:26 发布

yiyayi@che

最新推荐文章于 2025-07-19 10:02:26 发布

阅读量3.7k

点赞数 2

CC 4.0 BY-SA版权

文章标签：深度学习 pytorch 神经网络

本文链接：https://blog.csdn.net/yiyayi1/article/details/111314520

博客记录了TensorRT出现的错误，如‘invalid resource handle’和‘FAILED_EXECUTION’等，还给出了参考博客，并提及要通过修改部分代码来解决问题，与深度学习、神经网络相关。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[TensorRT] ERROR: ../rtSafe/cuda/reformat.cu (925) - Cuda Error in NCHWToNCHHW2: 400 (invalid resource handle)
[TensorRT] ERROR: FAILED_EXECUTION: std::exception
参考Python多线程中使用TensorRT_卑微小岳在线debug的博客-CSDN博客_tensorrt多线程

TensorRT python多进程推理踩坑_Mist-S的博客-CSDN博客_多线程推理

解决方法，修改以下部分代码

cuda_ctx = cuda.Device(0).make_context()   
self.cuda_ctx = cuda_ctx
if self.cuda_ctx:
    self.cuda_ctx.push()



if self.cuda_ctx:
    self.cuda_ctx.pop()