onnx 转 tensorRT报错解决

最新推荐文章于 2024-03-05 20:30:59 发布

日式炸猪排、日式炸只因

最新推荐文章于 2024-03-05 20:30:59 发布

阅读量3.3k

点赞数 2

文章标签： TensorRT CUDA 10.2 内核错误 patch补丁安装教程

本文链接：https://blog.csdn.net/weixin_41977938/article/details/121742602

版权

报错如下：

[12/06/2021-10:20:30] [E] Error[2]: [ltWrapper.cpp::setupHeuristic::327] Error Code 2: Internal Error (Assertion cublasStatus == CUBLAS_STATUS_SUCCESS failed.)
[12/06/2021-10:20:30] [E] Error[2]: [builder.cpp::buildSerializedNetwork::417] Error Code 2: Internal Error (Assertion enginePtr != nullptr failed.)
段错误 (核心已转储)

解决：

TensorRT8 cuda10.2下会出现这种问题

下载两个patch补丁

CUDA Toolkit 10.2 Download | NVIDIA Developer

下载完后，chmod 777 cuda_xxxxx.run 给权限

然后 ./cuda_xxxxx.run 直接运行就可以安装