错误信息:
E20211126 11:55:13.654 140412916315904 tensorrt.cpp:10] …/rtSafe/cuda/caskUtils.cpp (98) - Assertion Error in trtSmToCask: 0 (Unsupported SM.)
原因:TensorRT版本与cuda或者cudnn版本不匹配,如笔者的TensorRT版本为7.2.1,而cuda为最新的11.5版本,其中cuda版本过高无法适配:
解决:重装cuda或者TensorRT
建议:重装cuda,因为不同版本的TensorRT在使用时存在代码不一致的情况,如TensorRT7.2.1和TensorRT8.2.1要求重写虚函数时添加noexcept,如果代码量不大可以考虑重装TensorRT
附上TensorRT、cuda、cudnn各版本的匹配要求:TensorRT release note