diff --git a/training/deepspeed/Containerfile b/training/deepspeed/Containerfile
index 854864d..2b86e18 100644
--- a/training/deepspeed/Containerfile
+++ b/training/deepspeed/Containerfile
@@ -3,7 +3,7 @@ FROM nvcr.io/nvidia/cuda:12.1.1-cudnn8-devel-ubi9
 
 RUN dnf install -y python python-devel git
 RUN python -m ensurepip --upgrade
-RUN pip3 install torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
+RUN pip3 install torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121
 RUN pip3 install packaging wheel
 RUN pip3 install flash-attn==2.5.7
 RUN pip3 install deepspeed==0.14.2