Check this out (converting and running a pytorch model in a tensorRT engine) -->
https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/topics/topics/workflows/manually_construct_tensorrt_engine.html 821