Skip to content

Commit 14fa971

Browse files
author
Tyler Titsworth
authored
update pytorch cpu whl index (#312)
Signed-off-by: tylertitsworth <[email protected]>
1 parent 4a7f238 commit 14fa971

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

pytorch/requirements.txt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
torch==2.4.0
2-
torchvision==0.19.0
3-
torchaudio==2.4.0
4-
-f https://download.pytorch.org/whl/cpu/torch_stable.html
1+
torch==2.4.0+cpu
2+
torchvision==0.19.0+cpu
3+
torchaudio==2.4.0+cpu
4+
--extra-index-url https://download.pytorch.org/whl/cpu
55
intel_extension_for_pytorch==2.4.0+cpu
66
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/

pytorch/serving/build-kfs.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717
cd .. || exit
1818
docker compose pull torchserve
19-
docker tag "$(docker images -q | head -n1)" intel/torchserve:latest
19+
docker tag "${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-ubuntu-22.04-py3.10-torchserve" intel/torchserve:latest
2020
git clone https://github.com/pytorch/serve
2121
cd serve/kubernetes/kserve || exit
2222
git apply ../../../serving/kfs.patch

pytorch/serving/patch.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ spec:
242242
- grpc-v1
243243
containers:
244244
- name: kserve-container
245-
image: "intel/intel-extension-for-pytorch:2.2.0-serving-cpu-kserve"
245+
image: "intel/intel-extension-for-pytorch:2.4.0-serving-cpu-kserve"
246246
args:
247247
- torchserve
248248
- --start

0 commit comments

Comments
 (0)