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

[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

猜你喜欢

转载自blog.csdn.net/weixin_45306341/article/details/128485196