Update pytorch to 2.3.0

http://b/337048719

Author: djherbis

config.txt

@@ -3,10 +3,10 @@ BASE_IMAGE_TAG=m114
 CPU_BASE_IMAGE_NAME=tf2-cpu.2-15.py310
 GPU_BASE_IMAGE_NAME=tf2-gpu.2-15.py310
 LIGHTGBM_VERSION=4.2.0
-TORCH_VERSION=2.2.1
-TORCHAUDIO_VERSION=2.2.1
-TORCHTEXT_VERSION=0.17.1
-TORCHVISION_VERSION=0.17.1
+TORCH_VERSION=2.3.0
+TORCHAUDIO_VERSION=2.3.0
+TORCHTEXT_VERSION=0.18.0
+TORCHVISION_VERSION=0.18.0
 JAX_VERSION=0.4.26
 CUDA_MAJOR_VERSION=12
 CUDA_MINOR_VERSION=1