attp 9

Author: calderjo

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.3.1
+TORCHAUDIO_VERSION=2.3.1
+TORCHTEXT_VERSION=0.18.0
+TORCHVISION_VERSION=0.18.1
 JAX_VERSION=0.4.26
 CUDA_MAJOR_VERSION=12
 CUDA_MINOR_VERSION=3