Draft: Add support for TRT 10.X.X
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