TensorRT no longer has NvUtils.h - build from source is failing

[BAlmeidaS]

Issue type

Build/Install

Have you reproduced the bug with TensorFlow Nightly?

No

Source

source

TensorFlow version

tf 2.16.1

Custom code

Yes

OS platform and distribution

Linux Arch

Mobile device

No response

Python version

3.12.3

Bazel version

6.5.0

GCC/compiler version

14.1.1

CUDA/cuDNN version

12.4

GPU model and memory

RTX 4090 - 24gb

Current behavior?

While building from source, when I specify to use Tensorrt 10.0.1, the bazel configuration uses the header files from version 8 (https://github.com/tensorflow/tensorflow/blob/master/third_party/tensorrt/tensorrt_configure.bzl#L50-L65), which were changed in major 9 of Tensorrt (https://github.com/NVIDIA/TensorRT/tree/v9.1.0/include).

There is no longer NvUtils.h and a new header NvOnnxConfig.h (since 9.x.x and still like this on TensorRT 10).

I believe that either a new configuration must be created to support TensorRT 9.x.x and 10.x.x, or you should specify that Tensorflow only accepts TensortRT <= 8.6.

Standalone code to reproduce the issue

USE_BAZEL_VERSION=6.5.0 bazel build //tensorflow/tools/pip_package/v2:wheel --repo_env=WHEEL_NAME=tensorflow --config=cuda --jobs=30 --verbose_failuresj

Relevant log output

No response