Upgrade TENSORFLOW_VERSION env variable

To match the `tf2-gpu.2-6:m87` version of TensorFlow installed.

Author: rosbo

Dockerfile.tmpl

@@ -24,7 +24,7 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib
 FROM ${BASE_IMAGE_REPO}/${CPU_BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}
 {{ end }}
 # Keep these variables in sync if base image is updated.
-ENV TENSORFLOW_VERSION=2.6.0
+ENV TENSORFLOW_VERSION=2.6.2
 
 # We need to redefine the ARG here to get the ARG value defined above the FROM instruction.
 # See: https://docs.docker.com/engine/reference/builder/#understand-how-arg-and-from-interact