built-onnxruntime-for-raspberrypi-linux/build_parallel.sh at master · nknytk/built-onnxruntime-for-raspberrypi-linux · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#!/bin/bash

function build_onnx_runtime() {
  set -e
  onnxruntime_ver=$1
  pyver=$2
  work_dir="build_${onnxruntime_ver}_${pyver}"
  mkdir ${work_dir}
  cd ${work_dir}

  cat ../Dockerfile-arm32v7.template | sed "s/PYTHON_MINOR_VERSION/${pyver}/" | sed "s/ONNXRUNTIME_VERSION/${onnxruntime_ver}/" > Dockerfile
  docker build --rm=true -t onnxruntime-arm32v7-3${pyver}:${onnxruntime_ver} .

  set +e
  docker run --name distc -v $(pwd)/built_wheels:/built_wheels --entrypoint "cross-build-start" onnxruntime-arm32v7-3${pyver}:${onnxruntime_ver}
  filename1=onnxruntime-${onnxruntime_ver}-cp3${pyver}-cp3${pyver}m-linux_armv7l.whl
  filename2=onnxruntime-${onnxruntime_ver}-cp3${pyver}-cp3${pyver}-linux_armv7l.whl
  # for ubuntu 18.04 or later
  docker cp distc:/code/onnxruntime/build/Linux/MinSizeRel/dist/${filename1} /tmp
  docker cp distc:/code/onnxruntime/build/Linux/MinSizeRel/dist/${filename2} /tmp
  mv /tmp/snap.docker/tmp/${filename1} ../wheels/buster/
  mv /tmp/snap.docker/tmp/${filename2} ../wheels/buster/
  # for ubuntu 16.04 or later
  #docker cp distc:/code/onnxruntime/build/Linux/MinSizeRel/dist/${filename1} /tmp/
  #docker cp distc:/code/onnxruntime/build/Linux/MinSizeRel/dist/${filename2} /tmp/
  #mv /tmp/${filename1} ../wheels/buster/
  #mv /tmp/${filename2} ../wheels/buster/
  set -e

  docker rm distc
  docker rmi onnxruntime-arm32v7-3${pyver}:${onnxruntime_ver}

  cd ..
  rm -r ${work_dir}
}

v=1.4.0
build_onnx_runtime ${v} 5 & build_onnx_runtime ${v} 6 & build_onnx_runtime ${v} 7 & build_onnx_runtime ${v} 8