Merge pull request #74408 from lubinsz/pr_tf

node-perf/tf-wide-deep: add support for Arm64
k3s-v1.15.3
Kubernetes Prow Robot 2019-03-19 19:01:17 -07:00 committed by GitHub
commit 66674f5496
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 1 deletions

View File

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

View File

@ -14,8 +14,18 @@
FROM BASEIMAGE FROM BASEIMAGE
CROSS_BUILD_COPY qemu-QEMUARCH-static /usr/bin/
RUN apt-get update && apt-get install -y wget time 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 \ RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \
&& tar xzf v1.9.0.tar.gz \ && tar xzf v1.9.0.tar.gz \