mlflow-server/Dockerfile at main · lostless13/mlflow-server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
FROM python:3.10-slim

# Install MLflow and the two required drivers
# 1. psycopg2-binary: To talk to your PostgreSQL database
# 2. google-cloud-storage: To talk to your GCS bucket
RUN pip install mlflow \
                psycopg2-binary \
                google-cloud-storage

# Set the container port (Cloud Run's default)
EXPOSE 8080

# CMD mlflow gc --backend-store-uri ${MLFLOW_BACKEND_STORE_URI} \
#     --default-artifact-root ${MLFLOW_DEFAULT_ARTIFACT_ROOT}

CMD mlflow server \
    --host 0.0.0.0 \
    --port 8080 \
    --backend-store-uri ${MLFLOW_BACKEND_STORE_URI} \
    --default-artifact-root ${MLFLOW_DEFAULT_ARTIFACT_ROOT} \
    --no-serve-artifacts

# gcloud builds submit --tag us-central1-docker.pkg.dev/cam-triangle/elysian-docker/mlflow-server