Add tensorflow build for aarch64
This commit is contained in:
parent
3b18f8e883
commit
c536bdd029
3 changed files with 18 additions and 1 deletions
|
@ -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
|
||||
|
|
3
docker/tensorflow/build_dynamic.sh
Executable file
3
docker/tensorflow/build_dynamic.sh
Executable file
|
@ -0,0 +1,3 @@
|
|||
#!/bin/bash
|
||||
time bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so
|
||||
|
14
docker/tensorflow/docker-compose-aarch64.yml
Normal file
14
docker/tensorflow/docker-compose-aarch64.yml
Normal 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
|
||||
|
||||
|
Loading…
Reference in a new issue