diff --git a/sdk/python/kubeflow/trainer/Dockerfile b/sdk/python/kubeflow/trainer/Dockerfile
index d0ebee4aa3..9acea2a66b 100644
--- a/sdk/python/kubeflow/trainer/Dockerfile
+++ b/sdk/python/kubeflow/trainer/Dockerfile
@@ -1,5 +1,5 @@
 # Use an official Pytorch runtime as a parent image
-FROM nvcr.io/nvidia/pytorch:23.10-py3
+FROM nvcr.io/nvidia/pytorch:23.12-py3
 
 # Set the working directory in the container
 WORKDIR /app