Add cuda-nvml-dev-11-8 to Dockerfile.devel-gpu (and remove from Docke…

…rfile.gpu) PiperOrigin-RevId: 587829824
tensorflow · Dec 4, 2023 · 028aac5 · 028aac5
1 parent 4693db3
commit 028aac5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/tensorflow_serving/tools/docker/Dockerfile.devel-gpu b/tensorflow_serving/tools/docker/Dockerfile.devel-gpu
@@ -37,6 +37,7 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
         cuda-nvrtc-dev-11-8 \
         cuda-nvprune-11-8 \
         cuda-nvcc-11-8 \
+        cuda-nvml-dev-11-8 \
         cuda-cudart-dev-11-8 \
         libcufft-dev-11-8 \
         libcurand-dev-11-8 \

diff --git a/tensorflow_serving/tools/docker/Dockerfile.gpu b/tensorflow_serving/tools/docker/Dockerfile.gpu
@@ -36,7 +36,6 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
         ca-certificates \
         cuda-command-line-tools-11-8 \
         cuda-nvrtc-11-8 \
-        cuda-nvml-dev-11-8 \
         libcublas-11-8 \
         libcublas-dev-11-8 \
         libcufft-11-8 \