Pin libnvinfer headers versions. (#2212)

PiperOrigin-RevId: 620487661 Co-authored-by: Matt Miecnikowski <mattmiec@google.com>
tensorflow · Apr 5, 2024 · 6b8a8bc · 6b8a8bc
1 parent 36e151a
commit 6b8a8bc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 \