Add tensorflow build for aarch64

This commit is contained in:
Guy Sheffer 2020-01-17 14:10:23 +00:00
parent 3b18f8e883
commit c536bdd029
3 changed files with 18 additions and 1 deletions

View file

@ -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

View file

@ -0,0 +1,3 @@
#!/bin/bash
time bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so

View file

@ -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