[TensorRT] Cuda Error in allocate: 2

 

 

[TensorRT] ERROR: runtime.cpp (24) - Cuda Error in allocate: 2
[TensorRT] ERROR: runtime.cpp (24) - Cuda Error in allocate: 2

 

Cuda Memory 설정과 관련된 문제로서

 

builder.max_workspace_size  = 1 << 30

 

적절하게 설정해준다.