Skip to content

Draft: Add support for TRT 10.X.X

Nathan Thoumine requested to merge nthm/aidge_export_tensorrt:trt_10 into main

Context

This MR aims to add support for newer versions of TensorRT, either by creating aidge_export_tensorrt/static/tensorrt_10.X.X or by putting guards as in N2D2.

Modified files

Added tensorrt_10.5 folder which is for now just a copy of tensorrt_8.6 folder in static/ directory. The Makefile and Dockerfile have been modified to facilitate working with TRT 10.5.

Detailed major modifications

x

TODO

  • Modify the Makefile to enable the use of the latest TensorRT versions with Docker.
  • Identify and refactor deprecated code to maintain compatibility and best practices.
  • Provide users with the ability to select and use the desired TensorRT version. → Already exists in the export function via the trt_version parameter.

Issues

When exporting a model with the specified version 10.5, the src/ directory contains Dockerfiles for both versions 8.6 and 10.5:

import aidge_export_tensorrt

aidge_export_tensorrt.export("export", "model.onnx", trt_version= "10.5")
Edited by Nathan Thoumine

Merge request reports

Loading