报错如下:

[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 直接运行就可以安装

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐