fix build script

kubeflow · Aug 28, 2022 · f05cbf0 · f05cbf0
1 parent f10c5de
commit f05cbf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/v1beta1/build.sh b/scripts/v1beta1/build.sh
@@ -140,7 +140,7 @@ else
 fi
 
 echo -e "\nBuilding mxnet mnist training container example...\n"
-docker buildx build --platform linux/amd64 -t "${REGISTRY}/mxnet-mnist:${TAG}" -f examples/${VERSION}/trial-images/mxnet-mnist/Dockerfile .
+docker buildx build --platform "linux/$ARCH" -t "${REGISTRY}/mxnet-mnist:${TAG}" -f examples/${VERSION}/trial-images/mxnet-mnist/Dockerfile .
 
 echo -e "\nBuilding Tensorflow with summaries mnist training container example...\n"
 docker buildx build --platform "linux/$ARCH" -t "${REGISTRY}/tf-mnist-with-summaries:${TAG}" -f examples/${VERSION}/trial-images/tf-mnist-with-summaries/Dockerfile .