diff --git a/cmd/metricscollector/v1alpha3/tfevent-metricscollector/Dockerfile.ppc64le b/cmd/metricscollector/v1alpha3/tfevent-metricscollector/Dockerfile.ppc64le new file mode 100644 index 00000000000..b8cddd5260c --- /dev/null +++ b/cmd/metricscollector/v1alpha3/tfevent-metricscollector/Dockerfile.ppc64le @@ -0,0 +1,7 @@ +FROM ibmcom/tensorflow-ppc64le +RUN pip install rfc3339 grpcio googleapis-common-protos +ADD . /usr/src/app/github.com/kubeflow/katib +WORKDIR /usr/src/app/github.com/kubeflow/katib/cmd/metricscollector/v1alpha3/tfevent-metricscollector/ +RUN pip install --no-cache-dir -r requirements.txt +ENV PYTHONPATH /usr/src/app/github.com/kubeflow/katib:/usr/src/app/github.com/kubeflow/katib/pkg/apis/manager/v1alpha3/python:/usr/src/app/github.com/kubeflow/katib/pkg/metricscollector/v1alpha3/tfevent-metricscollector/:/usr/src/app/github.com/kubeflow/katib/pkg/metricscollector/v1alpha3/common/ +ENTRYPOINT ["python", "main.py"] diff --git a/scripts/v1alpha3/build.sh b/scripts/v1alpha3/build.sh index f0947e786b9..6fa7ab16836 100755 --- a/scripts/v1alpha3/build.sh +++ b/scripts/v1alpha3/build.sh @@ -61,6 +61,8 @@ docker build -t ${REGISTRY}/${PREFIX}/file-metrics-collector:${TAG} -f ${CMD_PRE echo "Building TF Event metrics collector image..." if [ $MACHINE_ARCH == "aarch64" ]; then docker build -t ${REGISTRY}/${PREFIX}/tfevent-metrics-collector:${TAG} -f ${CMD_PREFIX}/metricscollector/v1alpha3/tfevent-metricscollector/Dockerfile.aarch64 . +elif [ $MACHINE_ARCH == "ppc64le" ]; then + docker build -t ${REGISTRY}/${PREFIX}/tfevent-metrics-collector:${TAG} -f ${CMD_PREFIX}/metricscollector/v1alpha3/tfevent-metricscollector/Dockerfile.ppc64le . else docker build -t ${REGISTRY}/${PREFIX}/tfevent-metrics-collector:${TAG} -f ${CMD_PREFIX}/metricscollector/v1alpha3/tfevent-metricscollector/Dockerfile . fi