From c536bdd0296842ead16e7eed7dcc50e5beeb0bf8 Mon Sep 17 00:00:00 2001 From: Guy Sheffer Date: Fri, 17 Jan 2020 14:10:23 +0000 Subject: [PATCH] Add tensorflow build for aarch64 --- docker/tensorflow/Dockerfile.aarch64 | 2 +- docker/tensorflow/build_dynamic.sh | 3 +++ docker/tensorflow/docker-compose-aarch64.yml | 14 ++++++++++++++ 3 files changed, 18 insertions(+), 1 deletion(-) create mode 100755 docker/tensorflow/build_dynamic.sh create mode 100644 docker/tensorflow/docker-compose-aarch64.yml diff --git a/docker/tensorflow/Dockerfile.aarch64 b/docker/tensorflow/Dockerfile.aarch64 index fc376fa44..689014b29 100644 --- a/docker/tensorflow/Dockerfile.aarch64 +++ b/docker/tensorflow/Dockerfile.aarch64 @@ -60,6 +60,6 @@ RUN mv bazel-0.24.1-aarch64 /usr/local/bin/bazel && chmod 755 /usr/local/bin/baz WORKDIR "/home/tensorflow/tensorflow-1.14.0" COPY ./*.sh ./ COPY ./*.diff ./ -COPY ./.tf_configure.bazelrc .tf_configure.bazelrc +COPY ./tf_configure.bazelrc .tf_configure.bazelrc COPY ./Makefile.aarch64 Makefile RUN make patch diff --git a/docker/tensorflow/build_dynamic.sh b/docker/tensorflow/build_dynamic.sh new file mode 100755 index 000000000..2928525f5 --- /dev/null +++ b/docker/tensorflow/build_dynamic.sh @@ -0,0 +1,3 @@ +#!/bin/bash +time bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so + diff --git a/docker/tensorflow/docker-compose-aarch64.yml b/docker/tensorflow/docker-compose-aarch64.yml new file mode 100644 index 000000000..d62fafc05 --- /dev/null +++ b/docker/tensorflow/docker-compose-aarch64.yml @@ -0,0 +1,14 @@ +version: '3.6' + +services: + build: + build: + context: . + dockerfile: Dockerfile.aarch64 + container_name: tensorflow-build + tty: true + restart: always + volumes: + - ./out:/output + +