Merge pull request #1254 from Kaggle/upgrade-tf2.12

Upgrade to tf2.12 and CUDA 11.8

Author: djherbis

Dockerfile.tmpl

@@ -36,7 +36,7 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib
 {{ end }}
 
 # Keep these variables in sync if base image is updated.
-ENV TENSORFLOW_VERSION=2.11.0
+ENV TENSORFLOW_VERSION=2.12.0
 
 # 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

config.txt

@@ -1,11 +1,11 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
 BASE_IMAGE_TAG=m108
-CPU_BASE_IMAGE_NAME=tf2-cpu.2-11.py310
-GPU_BASE_IMAGE_NAME=tf2-gpu.2-11.py310
+CPU_BASE_IMAGE_NAME=tf2-cpu.2-12.py310
+GPU_BASE_IMAGE_NAME=tf2-gpu.2-12.py310
 LIGHTGBM_VERSION=3.3.2
 TORCH_VERSION=2.0.0
 TORCHAUDIO_VERSION=2.0.1
 TORCHTEXT_VERSION=0.15.1
 TORCHVISION_VERSION=0.15.1
 CUDA_MAJOR_VERSION=11
-CUDA_MINOR_VERSION=3
+CUDA_MINOR_VERSION=8