Fix k8s container BASE_IMAGE (#2808)

docker/build_image.sh creates a CPU image with tag 'latest-cpu'. Use that tag when building the k8s kserve image. Co-authored-by: Sean Bowman <[email protected]> Co-authored-by: lxning <[email protected]>
pytorch · Nov 28, 2023 · 709e743 · 709e743
1 parent 7baa8a6
commit 709e743
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kubernetes/kserve/build_image.sh b/kubernetes/kserve/build_image.sh
@@ -2,7 +2,7 @@
 
 MACHINE=cpu
 DOCKER_TAG="pytorch/torchserve-kfs:latest"
-BASE_IMAGE="pytorch/torchserve:latest"
+BASE_IMAGE="pytorch/torchserve:latest-cpu"
 DOCKER_FILE="Dockerfile"
 BUILD_NIGHTLY=false
 USE_CUSTOM_TAG=false