Unverified Commit 66674f54 authored by Kubernetes Prow Robot's avatar Kubernetes Prow Robot Committed by GitHub

Merge pull request #74408 from lubinsz/pr_tf

node-perf/tf-wide-deep: add support for Arm64
parents 046dcbd1 cd0e86b4
amd64=python:3.6-slim-stretch amd64=python:3.6-slim-stretch
arm64=arm64v8/python:3.6-slim-stretch
...@@ -14,8 +14,18 @@ ...@@ -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 \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment