mmdetection-to-tensorrt icon indicating copy to clipboard operation
mmdetection-to-tensorrt copied to clipboard

Published 20 hours ago • verified publisher icon grimoire

"Myelin Error in MyelinGraphContext" occur while I try to run trt model

Open zkccpro opened this issue 3 years ago • 0 comments

First, I test a mmdet demo on https://github.com/open-mmlab/mmdetection, and it works. Then I convert the .pth model to .trt model use mmdet2trt, and it works as well. But when I run .trt model use the method in readme.md, it has a ERROR as follow: [TensorRT] ERROR: myelin/myelinGraphContext.h (26) - Myelin Error in MyelinGraphContext: 1 (myelinVersionMismatch : Compiled against cuBLAS 11.3.0.0 but running against cuBLAS 11.2.1.0. ) terminate called after throwing an instance of 'nvinfer1::MyelinError' what(): std::exception Aborted (core dumped) My env are: CUDA11.2, cudnn8.0.4, TensorRT7.2.3.4, mmdet2.25.1. Now my confusion is: why I run .trt model ocuurs ERROR while the .pth model works properly?

zkccpro avatar Aug 22 '22 02:08 zkccpro