PyCUDA WARNING: a clean-up operation failed (dead context maybe?

Not L

于 2022-12-29 17:24:31 发布

阅读量825

点赞数 1

文章标签： linux

本文链接：https://blog.csdn.net/weixin_45306341/article/details/128485196

版权

[12/29/2022-17:14:07] [TRT] [E] 1: [cudaDriverHelpers.cpp::operator()::29] Error Code 1: Cuda Driver (an illegal memory access was encountered)
PyCUDA WARNING: a clean-up operation failed (dead context maybe?)
cuStreamDestroy failed: an illegal memory access was encountered
PyCUDA WARNING: a clean-up operation failed (dead context maybe?)
cuMemFree failed: an illegal memory access was encountered

trt推理过程中遇到上述问题
解决
模型输入或输出有多个，分配内存时应进行调整，或将模型输入输出调整为一个，参考
https://blog.csdn.net/irving512/article/details/115403888