Skip to content

Commit

Permalink
Updated onnxrt_build.sh (adding --allow_running_as_root) and Dockerfi…
Browse files Browse the repository at this point in the history
…le (pip install flatbuffers)
  • Loading branch information
junechul committed Nov 30, 2024
1 parent 9b39876 commit 7a4df55
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 4 deletions.
4 changes: 2 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -84,9 +84,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
python3 -m pip install --upgrade pip; \
fi && \
if echo ${BASE_IMAGE} | grep -q "debian" || echo ${BASE_IMAGE} | grep -q "ubuntu:24.04"; then \
python3 -m pip install --break-system-packages numpy setuptools wheel pybind11 pytest; \
python3 -m pip install --break-system-packages numpy setuptools wheel pybind11 pytest flatbuffers; \
else \
python3 -m pip install numpy setuptools wheel pybind11 pytest; \
python3 -m pip install numpy setuptools wheel pybind11 pytest flatbuffers; \
fi && \
rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*

Expand Down
3 changes: 2 additions & 1 deletion scripts/onnxrt_build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,8 @@ SECONDS=0
--cmake_extra_defines="CMAKE_TOOLCHAIN_FILE=$(pwd)/tool.cmake" \
--path_to_protoc_exe $(pwd)/cmake/external/protoc-${protobuf_ver}-linux-aarch_64/bin/protoc \
--use_tidl \
--build_wheel
--build_wheel \
--allow_running_as_root

# chmod
chmod -R a+w $WORK_DIR/workarea
Expand Down
2 changes: 1 addition & 1 deletion scripts/onnxrt_package.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ cd $WORK_DIR/workarea
## package into a tarball
onnx_ver=$(get_yaml_value "onnxruntime" "onnx_ver")
pkg_dist=${BASE_IMAGE//:/}
DST_DIR="onnx-${onnx_ver}+${tidl_ver}-${pkg_dist}_aarch64"
DST_DIR="onnx-${onnx_ver}-${pkg_dist}_aarch64"
LIB_DIR="onnxruntime/build/Linux/Release"
TARBALL="${DST_DIR}.tar.gz"

Expand Down

0 comments on commit 7a4df55

Please sign in to comment.