node-perf/tf-wide-deep: add support for Arm64

Signed-off-by: Bin Lu <bin.lu@arm.com>
This commit is contained in:
Bin Lu 2019-02-22 17:40:59 +08:00
parent 8dd09e0b36
commit cd0e86b4fc
2 changed files with 12 additions and 1 deletions

View File

@ -1 +1,2 @@
amd64=python:3.6-slim-stretch
arm64=arm64v8/python:3.6-slim-stretch

View File

@ -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 \