weixin_47911596 2023-03-08 11:27 采纳率: 0%
浏览 522
已结题

[TRT] [E] 1: [context.cpp::setStream::121] Error Code 1: Cudnn (CUDNN_STATUS_MAPPING_ERROR)

在使用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版本也更换了好几次,依然没有解决,想请了解的能指导一二,万分感谢!

  • 写回答

6条回答 默认 最新

  • 「已注销」 2023-03-08 11:30
    关注

    “Devil组”引证GPT后的撰写:

    • 这个错误通常是由于使用的cuDNN版本与TensorRT版本不兼容导致的。虽然已经更换了cudnn版本,但是可能还需要检查TensorRT和CUDA版本是否与您使用的cuDNN版本兼容。
    评论

报告相同问题?

问题事件

  • 系统已结题 3月16日
  • 创建了问题 3月8日