diff --git a/pytorch/requirements.txt b/pytorch/requirements.txt
index 33202d787..664b5ad8b 100644
--- a/pytorch/requirements.txt
+++ b/pytorch/requirements.txt
@@ -1,6 +1,6 @@
-torch==2.4.0
-torchvision==0.19.0
-torchaudio==2.4.0
--f https://download.pytorch.org/whl/cpu/torch_stable.html
+torch==2.4.0+cpu
+torchvision==0.19.0+cpu
+torchaudio==2.4.0+cpu
+--extra-index-url https://download.pytorch.org/whl/cpu
 intel_extension_for_pytorch==2.4.0+cpu
 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/
diff --git a/pytorch/serving/build-kfs.sh b/pytorch/serving/build-kfs.sh
index 7cdedc934..89e308232 100755
--- a/pytorch/serving/build-kfs.sh
+++ b/pytorch/serving/build-kfs.sh
@@ -16,7 +16,7 @@
 
 cd .. || exit
 docker compose pull torchserve
-docker tag "$(docker images -q | head -n1)" intel/torchserve:latest
+docker tag "${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-ubuntu-22.04-py3.10-torchserve" intel/torchserve:latest
 git clone https://github.com/pytorch/serve
 cd serve/kubernetes/kserve || exit
 git apply ../../../serving/kfs.patch
diff --git a/pytorch/serving/patch.yaml b/pytorch/serving/patch.yaml
index 487eab342..cd18e8470 100644
--- a/pytorch/serving/patch.yaml
+++ b/pytorch/serving/patch.yaml
@@ -242,7 +242,7 @@ spec:
     - grpc-v1
   containers:
     - name: kserve-container
-      image: "intel/intel-extension-for-pytorch:2.2.0-serving-cpu-kserve"
+      image: "intel/intel-extension-for-pytorch:2.4.0-serving-cpu-kserve"
       args:
         - torchserve
         - --start