adfdsafasdfsdaf

Author: calderjo

Dockerfile.tmpl

@@ -209,10 +209,6 @@ RUN pip install \
         tf-keras && \
     /tmp/clean-layer.sh
 
-# b/318672158 Use simply tensorflow_decision_forests on next release, expected with tf 2.16
-RUN pip install tensorflow_decision_forests==1.8.1 --no-deps && \
-    /tmp/clean-layer.sh
-
 RUN chmod +x /tmp/keras_patch.sh && \
     /tmp/keras_patch.sh
 

config.txt

@@ -3,10 +3,10 @@ BASE_IMAGE_TAG=m122
 CPU_BASE_IMAGE_NAME=tf2-cpu.2-16.py310
 GPU_BASE_IMAGE_NAME=tf2-gpu.2-16.py310
 LIGHTGBM_VERSION=4.2.0
-TORCH_VERSION=2.1.2
-TORCHAUDIO_VERSION=2.1.2
-TORCHTEXT_VERSION=0.16.2
-TORCHVISION_VERSION=0.16.2
+TORCH_VERSION=2.4.0
+TORCHAUDIO_VERSION=2.4.0
+TORCHTEXT_VERSION=0.18.0
+TORCHVISION_VERSION=0.19.0
 JAX_VERSION=0.4.26
 CUDA_MAJOR_VERSION=12
 CUDA_MINOR_VERSION=3