Fix gpu docker build failure due to bad substitution

PiperOrigin-RevId: 457141259
tensorflow · Jun 25, 2022 · 1d7cd5b · 1d7cd5b
1 parent 5c4a952
commit 1d7cd5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_serving/tools/docker/Dockerfile.gpu b/tensorflow_serving/tools/docker/Dockerfile.gpu
@@ -34,7 +34,7 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
     apt-get update && apt-get install -y --no-install-recommends \
         ca-certificates \
         cuda-command-line-tools-11-2 \
-        cuda-nvrtc-${CUDA/./-} \
+        cuda-nvrtc-11-2 \
         libcublas-11-2 \
         libcublas-dev-11-2 \
         libcufft-11-2 \