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 2e65f0fba2468..ba1103dd1ef96 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==1.13.1
+torch==2.2.0
 numpy==1.22.4
 Pillow==10.2.0
\ No newline at end of file