enhance armnn conversion
This commit is contained in:
28
machine-learning/export/ann/Dockerfile
Normal file
28
machine-learning/export/ann/Dockerfile
Normal file
@@ -0,0 +1,28 @@
|
||||
FROM mambaorg/micromamba:bookworm-slim@sha256:333f7598ff2c2400fb10bfe057709c68b7daab5d847143af85abcf224a07271a as builder
|
||||
|
||||
ENV TRANSFORMERS_CACHE=/cache \
|
||||
PYTHONDONTWRITEBYTECODE=1 \
|
||||
PYTHONUNBUFFERED=1 \
|
||||
PATH="/opt/venv/bin:$PATH"
|
||||
|
||||
WORKDIR /export/ann
|
||||
|
||||
USER root
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
build-essential \
|
||||
curl \
|
||||
git
|
||||
|
||||
USER $MAMBA_USER
|
||||
COPY --chown=$MAMBA_USER:$MAMBA_USER env.yaml ./
|
||||
RUN micromamba install -y -f env.yaml
|
||||
COPY --chown=$MAMBA_USER:$MAMBA_USER *.sh *.cpp ./
|
||||
|
||||
ENV ARMNN_PATH=/export/ann/armnn
|
||||
RUN ./download-armnn.sh && \
|
||||
./build-converter.sh && \
|
||||
./build.sh
|
||||
COPY --chown=$MAMBA_USER:$MAMBA_USER run.py ./
|
||||
|
||||
ENTRYPOINT ["/usr/local/bin/_entrypoint.sh"]
|
||||
CMD ["python", "run.py"]
|
||||
Reference in New Issue
Block a user