wehub-resource-sync
52 lines
2.1 KiB
Plaintext
52 lines
2.1 KiB
Plaintext
# Build an image that can serve mlflow models.
|
|
FROM ubuntu:22.04
|
|
|
|
RUN apt-get -y update && DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get install -y --no-install-recommends wget curl nginx ca-certificates bzip2 build-essential cmake git-core
|
|
|
|
# Setup pyenv
|
|
RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get -y install tzdata \
|
|
libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
|
|
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev
|
|
RUN git clone \
|
|
--depth 1 \
|
|
--branch $(git ls-remote --tags --sort=v:refname https://github.com/pyenv/pyenv.git | grep -o -E 'v[1-9]+(\.[1-9]+)+$' | tail -1) \
|
|
https://github.com/pyenv/pyenv.git /root/.pyenv
|
|
ENV PYENV_ROOT="/root/.pyenv"
|
|
ENV PATH="$PYENV_ROOT/bin:$PATH"
|
|
RUN apt install -y software-properties-common \
|
|
&& apt update \
|
|
&& add-apt-repository -y ppa:deadsnakes/ppa \
|
|
&& apt update \
|
|
&& apt install -y python3.10 python3.10-distutils \
|
|
# Remove python3-blinker to avoid pip uninstall conflicts
|
|
&& apt remove -y python3-blinker \
|
|
&& ln -s -f $(which python3.10) /usr/bin/python \
|
|
&& wget https://bootstrap.pypa.io/get-pip.py -O /tmp/get-pip.py \
|
|
&& python /tmp/get-pip.py
|
|
|
|
|
|
# Setup Java
|
|
RUN apt-get install -y --no-install-recommends openjdk-17-jdk maven
|
|
ENV JAVA_HOME=/usr/lib/jvm/java-17-openjdk-amd64
|
|
|
|
WORKDIR /opt/mlflow
|
|
|
|
# Install MLflow
|
|
RUN pip install ${{ MLFLOW_INSTALL }}
|
|
|
|
# Copy model to image and install dependencies
|
|
COPY model_dir/model /opt/ml/model
|
|
RUN python -c "from mlflow.models import container as C; C._install_pyfunc_deps('/opt/ml/model', install_mlflow=False, env_manager='virtualenv');"
|
|
|
|
ENV MLFLOW_DISABLE_ENV_CREATION=True
|
|
|
|
# granting read/write access and conditional execution authority to all child directories
|
|
# and files to allow for deployment to AWS Sagemaker Serverless Endpoints
|
|
# (see https://docs.aws.amazon.com/sagemaker/latest/dg/serverless-endpoints.html)
|
|
RUN chmod o+rwX /opt/mlflow/
|
|
|
|
# clean up apt cache to reduce image size
|
|
RUN rm -rf /var/lib/apt/lists/*
|
|
|
|
ENTRYPOINT ["python", "-c", "from mlflow.models import container as C; C._serve('virtualenv')"]
|