diff --git a/sdks/python/apache_beam/examples/ml-orchestration/kfp/components/train/requirements.txt b/sdks/python/apache_beam/examples/ml-orchestration/kfp/components/train/requirements.txt
index ba1103dd1ef9..5b996fdfc70a 100644
--- a/sdks/python/apache_beam/examples/ml-orchestration/kfp/components/train/requirements.txt
+++ b/sdks/python/apache_beam/examples/ml-orchestration/kfp/components/train/requirements.txt
@@ -13,6 +13,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-torch==2.2.0
+torch==2.12.0
 numpy==1.22.4
 Pillow==10.2.0
\ No newline at end of file