Add cuda option to TVM cli docker build script

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

Closed luca-fancellu closed 3 years ago

luca-fancellu commented 3 years ago

Add option to build cuda enabled TVM cli docker images

Add -c option to the build.sh script to build the docker image with cuda enabled.

Signed-off-by: Luca Fancellu luca.fancellu@arm.com Change-Id: Idd513bb1418c91a5ffdcd80c292ac85151b69c28