Skip to content

Commit 66674f5

Browse files
authored
Merge pull request kubernetes#74408 from lubinsz/pr_tf
node-perf/tf-wide-deep: add support for Arm64
2 parents 046dcbd + cd0e86b commit 66674f5

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)