Pin libnvinfer headers versions. (#2212) (#2218)

PiperOrigin-RevId: 620487661 Co-authored-by: Matt Miecnikowski <[email protected]>
tensorflow · Apr 12, 2024 · 0d72695 · 0d72695
1 parent 0e62613
commit 0d72695
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 \