在使用tensorrt部署yolov5时运行到context.execute_async(batch_size=self.batch_size, bindings=bindings, stream_handle=stream.handle)报错: [TRT] [E] 1: [context.cpp::setStream::121] Error Code 1: Cudnn (CUDNN_STATUS_MAPPING_ERROR)
我的系统是ubuntu20,显卡为rtx3060,tensorrt版本是8.4.3.1,cuda版本是11.6,cudnn采用的是8.4.2。一开始遇到这个错误时以为是cudnn版本不对,按照下载时的推荐更换了好多版本,包括cuda版本也更换了好几次,依然没有解决,想请了解的能指导一二,万分感谢!
[TRT] [E] 1: [context.cpp::setStream::121] Error Code 1: Cudnn (CUDNN_STATUS_MAPPING_ERROR)
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-
6条回答 默认 最新
- 「已注销」 2023-03-08 11:30关注
“Devil组”引证GPT后的撰写:
- 这个错误通常是由于使用的cuDNN版本与TensorRT版本不兼容导致的。虽然已经更换了cudnn版本,但是可能还需要检查TensorRT和CUDA版本是否与您使用的cuDNN版本兼容。
解决 无用评论 打赏 举报