Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Mlflow update 2209 #367

Merged
11 commits merged into from
Sep 22, 2022
Merged

Mlflow update 2209 #367

Show file tree
Hide file tree
Changes from 4 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
de5b9d1
Update mlflow with some extras
pdmack Aug 31, 2022
593ae00
Update mlflow with some extras
pdmack Aug 31, 2022
f13a99a
Merge branch 'mlflow-update-2209' of github.com:pdmack/Morpheus into …
pdmack Sep 20, 2022
1268f92
Merge branch 'branch-22.09' into mlflow-update-2209
pdmack Sep 20, 2022
1ea0726
Add fixes and clarity to mlflow readme
pdmack Sep 20, 2022
d754b42
Merge branch 'mlflow-update-2209' of github.com:pdmack/Morpheus into …
pdmack Sep 20, 2022
f32fa26
Merge branch 'branch-22.09' into mlflow-update-2209
pdmack Sep 21, 2022
e032723
Update mlflow-env.yml
pdmack Sep 21, 2022
dba1736
Update mlflow-env.yml
pdmack Sep 21, 2022
427bd4f
Some tweaks to line up with DFP mlflow
pdmack Sep 22, 2022
1646adc
Merge branch 'mlflow-update-2209' of github.com:pdmack/Morpheus into …
pdmack Sep 22, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions models/mlflow/docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,8 @@
# limitations under the License.

ARG FROM_IMAGE=gpuci/miniforge-cuda
ARG CUDA_VER=11.4
ARG CUDA_VER=11.5
ARG LINUX_VER=ubuntu20.04
ARG TRITON_VER=r22.02
FROM ${FROM_IMAGE}:${CUDA_VER}-runtime-${LINUX_VER} AS base

WORKDIR /mlflow
Expand All @@ -31,7 +30,7 @@ RUN sed -i 's/conda activate base/conda activate mlflow/g' ~/.bashrc
SHELL ["/opt/conda/bin/conda", "run", "-n", "mlflow", "/bin/bash", "-c"]

ARG TRITON_DIR=/mlflow/triton-inference-server
ARG TRITON_VER=r21.12
ARG TRITON_VER=r22.07

RUN mkdir ${TRITON_DIR} && \
cd ${TRITON_DIR} && \
Expand All @@ -43,6 +42,9 @@ RUN mkdir ${TRITON_DIR} && \
RUN ln -sf ${TRITON_DIR}/server/deploy/mlflow-triton-plugin/scripts/publish_model_to_mlflow.py /mlflow && \
mkdir /mlflow/artifacts

# remedy for CVE-2015-20107
RUN find / -name '*mailcap*.*py*' | xargs rm

ENV MLFLOW_MODEL_REPO=/mlflow/artifacts
ENV MLFLOW_TRACKING_URI=sqlite:////tmp/mlflow-db.sqlite

Expand Down
5 changes: 3 additions & 2 deletions models/mlflow/docker/conda/mlflow-env.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,8 @@ dependencies:
- python=3.8
- pip
- onnx
- openssl>=1.1.1o
- openssl>=1.1.1q
- psycopg2<3
- pip:
- boto3
- mlflow