diff --git a/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt b/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt
index f896d6c..6caf4be 100644
--- a/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt
+++ b/Container-Root/hyperpod/deployment/eks/demo/hero/training/fsdp/src/requirements.txt
@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cu128
 datasets
-torch==2.7.1
+torch==2.7.1+cu128
 torchaudio==2.7.1
 torchvision==0.22.1
 transformers==4.52.4