Skip to content

Commit cd0e86b

Browse files
author
Bin Lu
committed
node-perf/tf-wide-deep: add support for Arm64
Signed-off-by: Bin Lu <[email protected]>
1 parent 8dd09e0 commit cd0e86b

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
11
amd64=python:3.6-slim-stretch
2+
arm64=arm64v8/python:3.6-slim-stretch

test/images/node-perf/tf-wide-deep/Dockerfile

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,18 @@
1414

1515
FROM BASEIMAGE
1616

17+
CROSS_BUILD_COPY qemu-QEMUARCH-static /usr/bin/
18+
1719
RUN apt-get update && apt-get install -y wget time
18-
RUN pip install tensorflow
20+
21+
RUN case $(uname -m) in \
22+
aarch64) \
23+
pip install tensorflow-aarch64; \
24+
;; \
25+
*) \
26+
pip install tensorflow; \
27+
;; \
28+
esac
1929

2030
RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \
2131
&& tar xzf v1.9.0.tar.gz \

0 commit comments

Comments
 (0)