Upgrade TENSORFLOW_VERSION env variable

To match the `tf2-gpu.2-6:m87` version of TensorFlow installed.
Kaggle · Dec 16, 2021 · b47cb41 · b47cb41
1 parent cb4a134
commit b47cb41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.tmpl b/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