mmdeploy: RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

Hi, I am using mmdeploy to convert maskrcnn pth model to tensorrt, and i got this error: RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

my command: python tools/deploy.py \ configs/mmdet/instance-seg/instance-seg_tensorrt_dynamic-320x320-1344x1344.py \ /home/mmlab/mmdetection-2.20.0/project/mrcnn_r18_fpn_160e/mrcnn_r18_fpn_160e.py \ /home/mmlab/mmdetection-2.20.0/project/mrcnn_r18_fpn_160e/epoch_150.pth \ /home/mmlab/mmdetection-2.20.0/demo/demo.jpg \ --work-dir work_dir \ --device cuda:0

my env: gcc g++ 7.5.0 cuda 10.2 cudnn 7.6.5 tensorrt 7.2.3 torch 1.8.0

About this issue

  • Original URL
  • State: closed
  • Created 2 years ago
  • Comments: 16 (7 by maintainers)

Commits related to this issue

Most upvoted comments

You are welcome. Feel free to open issues or pull requests to MMDeploy in the future.

Need to clone the third-party library. Follow build. Well, reading the readme of this repository first is recommended.

Thank you for your help, my environment has been set up successfully !!!