Hi raul_91,
Yes, officially TensorRT Python API isn’t supported on Drive platforms as description in the doc.
Could you convert your model into UFF/ONNX one offline, and then import and run it with C++ API on your Drive system?
BTW, Please see if “Optimizing DNN Inference Using CUDA and TensorRT on NVIDIA DRIVE AGX” webinar, mentioned in below links, is helpful.
https://news.developer.nvidia.com/how-drive-agx-cuda-and-tensorrt-achieve-fast-accurate-autonomous-vehicle-perception/
https://devtalk.nvidia.com/default/topic/1064456/general/nvidia-webinars-mdash-optimizing-dnn-inference-using-cuda-and-tensorrt-on-nvidia-drive-agx/