search

autowarefoundation / modelzoo

A collection of machine-learned models for use in autonomous driving applications.
https://www.autoware.org/
Apache License 2.0
62 stars 24 forks source link

Dockerfile for aarch64 tvm cli docker #15

Closed LiyouZhou closed 3 years ago

LiyouZhou commented 3 years ago

On Aarch64 ubuntu, onnx and tensorflow pre-compiled packages are not provided. They need to be compiled from source.

The docker build is now split into 2 stages. The first stage is different between amd64 and arm64. A script build.sh is provided to orchestrate the build and hide this complexity from user.

Change-Id: If463859cb19761240ccaac0c2e19de4191017c1b Issue-Id: SCM-1463 Signed-off-by: Liyou Zhou liyou.zhou@arm.com

LiyouZhou commented 3 years ago

@luca-fancellu FYI

LiyouZhou commented 3 years ago

@JWhitleyWork Sorry it took so long to turn around. It was my fault for not testing thoroughly before submitting. I have made sure this time to test both on x86 and aarch64. If you don't mind, please have a look.

JWhitleyWork commented 3 years ago

@JWhitleyWork Sorry it took so long to turn around. It was my fault for not testing thoroughly before submitting. I have made sure this time to test both on x86 and aarch64. If you don't mind, please have a look.

Not a problem! This actually exposes a need on this repository (in my opinion): CI jobs for the scripts folder. I've created #17 to track this.

As for this PR, I've verified functionality on amd64 and arm64. LGTM!