diff --git a/config.txt b/config.txt
index 0a0a98fd..9bef3a48 100644
--- a/config.txt
+++ b/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.1.2
-TORCHAUDIO_VERSION=2.1.2
-TORCHTEXT_VERSION=0.16.2
-TORCHVISION_VERSION=0.16.2
+TORCH_VERSION=2.2.1
+TORCHAUDIO_VERSION=2.2.1
+TORCHTEXT_VERSION=0.17.1
+TORCHVISION_VERSION=0.17.1
 JAX_VERSION=0.4.23
 CUDA_MAJOR_VERSION=12
 CUDA_MINOR_VERSION=1