tfevent-metricscollector support ppc64le

kubeflow · Nov 5, 2019 · 1491d9b · 1491d9b
1 parent c18bab6
commit 1491d9b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/cmd/metricscollector/v1alpha3/tfevent-metricscollector/Dockerfile.ppc64le 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
@@ -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