From 0d726954f36a2b6c6bfb60a4caec311f2b3e3101 Mon Sep 17 00:00:00 2001 From: Raviteja Gorijala <36429068+rtg0795@users.noreply.github.com> Date: Fri, 12 Apr 2024 15:08:35 -0700 Subject: [PATCH] Pin libnvinfer headers versions. (#2212) (#2218) PiperOrigin-RevId: 620487661 Co-authored-by: Matt Miecnikowski --- tensorflow_serving/tools/docker/Dockerfile.devel-gpu | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow_serving/tools/docker/Dockerfile.devel-gpu b/tensorflow_serving/tools/docker/Dockerfile.devel-gpu index ee4a91a3356..15ff459b659 100644 --- a/tensorflow_serving/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow_serving/tools/docker/Dockerfile.devel-gpu @@ -56,7 +56,9 @@ RUN apt-get install -y --no-install-recommends \ libnvinfer-plugin8=8.6.1.6-1+cuda12.0 \ libnvinfer8=8.6.1.6-1+cuda12.0 \ libnvinfer-dev=8.6.1.6-1+cuda12.0 \ + libnvinfer-headers-dev=8.6.1.6-1+cuda12.0 \ libnvinfer-plugin-dev=8.6.1.6-1+cuda12.0 \ + libnvinfer-headers-plugin-dev=8.6.1.6-1+cuda12.0 \ automake \ build-essential \ ca-certificates \