root@shhjj-System-Product-Name:/home/shhjj/gpu/tensorrtx_yolov5_person-v6/build# ./yolov5 -s ../model/0826_ts_person_last.wts ../model/0826_ts_person_last.engine s
Loading weights: ../model/0826_ts_person_last.wts
Building engine, please wait for a while...
[10/31/2023-17:21:29] [E] [TRT] 1: [ltWrapper.cpp::CublasLtWrapper::66] Error Code 1: Cublas (Could not initialize cublasLt. Please check CUDA installation.)
Build engine successfully!
yolov5: /home/shhjj/gpu/tensorrtx_yolov5_person-v6/yolov5.cpp:243: void APIToModel(unsigned int, nvinfer1::IHostMemory**, bool&, float&, float&, std::string&): Assertion `engine != nullptr' failed.
已放弃 (核心已转储)
解决方法:
内存不足,看有没有其他程序占用cuda进程,kill掉 再跑一次 就成功了