Hi,
You can limit the TensorRT memory usage by setting workspace variable. Please check this comment for more information:
Thanks.