start building engine
[08/03/2022-23:18:04] [W] [TRT] Half2 support requested on hardware without native FP16 support, performance will be negatively affected.
[08/03/2022-23:18:05] [E] [TRT] ../rtSafe/safeRuntime.cpp (25) - Cuda Error in allocate: 2 (out of memory)
[08/03/2022-23:18:05] [E] [TRT] ../rtSafe/safeRuntime.cpp (25) - Cuda Error in allocate: 2 (out of memory)
build engine done
mmpose_trt: /home/pcb/Algorithm/tensorrt_inference/code/src/model.cpp:46: void Model::OnnxToTRTModel(): Assertion `engine' failed.
已放弃 (核心已转储)
./mmpose_trt ../../../configs/mmpose/config.yaml ../../../samples/pedestrian
Input filename: ../hrnet_w48_coco_256x192.onnx ONNX IR version: 0.0.6 Opset version: 11 Producer name: pytorch Producer version: 1.7 Domain:
Model version: 0 Doc string:
start building engine [08/03/2022-23:18:04] [W] [TRT] Half2 support requested on hardware without native FP16 support, performance will be negatively affected. [08/03/2022-23:18:05] [E] [TRT] ../rtSafe/safeRuntime.cpp (25) - Cuda Error in allocate: 2 (out of memory) [08/03/2022-23:18:05] [E] [TRT] ../rtSafe/safeRuntime.cpp (25) - Cuda Error in allocate: 2 (out of memory) build engine done mmpose_trt: /home/pcb/Algorithm/tensorrt_inference/code/src/model.cpp:46: void Model::OnnxToTRTModel(): Assertion `engine' failed. 已放弃 (核心已转储)
老哥你解决了么？我也遇到这个问题了