使用TensorRT的一些没注意到的坑

本文主要是记录平时遇到的一些奇奇怪怪的错误的

(1)编译生成的trt计划文件,在设备更换cudnn版本后,爆出如下错误:

error: 1: [convolutionrunner.cpp::nvinfer1::rt::cuda::cudnnconvolutionrunner::executeconv::458] error code 1: cudnn (cudnn_status_bad_param)

解决方法:重新生成engine文件后就可以了!

感想:以前我们都只知道生成的engine所用的cuda和部署到应用现场的设备的cuda版本保持一致或者都是相同架构和算力的,没想到同样的cuda版本,同一台设备,居然cudnn版本也会影响能否正常使用!

待继续更新。。。。。。。

猜你喜欢

转载自blog.csdn.net/yangjinyi1314/article/details/129423489
今日推荐