diff --git a/test/images/node-perf/tf-wide-deep/BASEIMAGE b/test/images/node-perf/tf-wide-deep/BASEIMAGE index 70bc384f0a..1e73049b04 100644 --- a/test/images/node-perf/tf-wide-deep/BASEIMAGE +++ b/test/images/node-perf/tf-wide-deep/BASEIMAGE @@ -1 +1,2 @@ amd64=python:3.6-slim-stretch +arm64=arm64v8/python:3.6-slim-stretch diff --git a/test/images/node-perf/tf-wide-deep/Dockerfile b/test/images/node-perf/tf-wide-deep/Dockerfile index f5ab2c5c8d..45e3e2496e 100644 --- a/test/images/node-perf/tf-wide-deep/Dockerfile +++ b/test/images/node-perf/tf-wide-deep/Dockerfile @@ -14,8 +14,18 @@ FROM BASEIMAGE +CROSS_BUILD_COPY qemu-QEMUARCH-static /usr/bin/ + RUN apt-get update && apt-get install -y wget time -RUN pip install tensorflow + +RUN case $(uname -m) in \ + aarch64) \ + pip install tensorflow-aarch64; \ + ;; \ + *) \ + pip install tensorflow; \ + ;; \ + esac RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \ && tar xzf v1.9.0.tar.gz \