dhi.io/mlflow
MLflow is an open-source platform for managing the machine learning lifecycle: experiment tracking, reproducible runs, model packaging, and deployment. This image provides the mlflow CLI and server runtime packaged as a hardened container for secure production deployments.
All examples in this guide use the public image. If you’ve mirrored the repository for your own use (for example, to your Docker Hub namespace), update your commands to reference the mirrored image instead of the public one.
For example:
dhi.io/<repository>:<tag><your-namespace>/dhi-<repository>:<tag>For the examples, you must first use docker login dhi.io to authenticate to the registry to pull the images.
This Docker Hardened mlflow image includes:
Below are common ways to run the mlflow server from the hardened image. The mlflow server exposes the tracking UI and API (default port 5000).
This image uses mlflow as its runtime entrypoint. Pass mlflow subcommands directly in docker run and Docker Compose
examples:
server, not mlflow servermodels build-docker, not mlflow models build-docker$ docker run -d --name mlflow-server -p 5000:5000 \
dhi.io/mlflow:3 \
server \
--backend-store-uri sqlite:///mlflow.db \
--default-artifact-root file:///opt/mlflow/data/mlruns \
--host 0.0.0.0 \
--port 5000
This starts a simple single-node tracking server with a local SQLite backend and local artifact folder (suitable for development and evaluation only).
services:
postgres:
image: dhi.io/postgres:18
environment:
POSTGRES_USER: mlflow
POSTGRES_PASSWORD: mlflow
POSTGRES_DB: mlflowdb
volumes:
- mlflow-db:/var/lib/postgresql/data
minio:
image: dhi.io/minio:0
environment:
MINIO_ROOT_USER: minioadmin
MINIO_ROOT_PASSWORD: minioadmin
command:
- server
- /data
ports:
- "9000:9000"
volumes:
- mlflow-minio:/data
mlflow:
image: dhi.io/mlflow:3
container_name: mlflow-server
ports:
- "5000:5000"
environment:
AWS_ACCESS_KEY_ID: minioadmin
AWS_SECRET_ACCESS_KEY: minioadmin
MLFLOW_S3_ENDPOINT_URL: http://minio:9000
command:
- server
- --backend-store-uri
- postgresql+psycopg2://mlflow:mlflow@postgres:5432/mlflowdb
- --default-artifact-root
- s3://mlflow-artifacts
- --host
- 0.0.0.0
- --port
- "5000"
depends_on:
- postgres
- minio
volumes:
mlflow-db:
mlflow-minio:
Basic single-node tracking server (development): SQLite backend and local artifact root. Not suitable for production.
Production tracking server: Postgres (or other SQL DB) as backend store and S3/MinIO as artifact store. Run behind a reverse proxy (nginx) for TLS and authentication.
Model packaging and serving: The models subcommands are available in the image. For detailed workflows and model
deployment patterns, refer to the upstream MLflow documentation at mlflow.org.
The hardened mlflow image focuses on security (minimal runtime, nonroot user, reduced packages). The main migration
difference is that the runtime image uses mlflow as its entrypoint, so container commands must pass mlflow subcommands
directly.
Docker Hardened Images come in different variants depending on their intended use. Image variants are identified by their tag.
Runtime variants are designed to run your application in production. These images are intended to be used either directly or as the FROM image in the final stage of a multi-stage build. These images typically:
Build-time variants typically include dev in the tag name and are intended for use in the first stage of a
multi-stage Dockerfile. These images typically:
To view the image variants and get more information about them, select the Tags tab for this repository, and then select a tag.
fips in the tag. They layer the OpenSSL FIPS provider and a dhi/python runtime tagged for
FIPS (-debian13-fips) so TLS and other OpenSSL-backed crypto use FIPS-validated modules. Application-level Python
cryptography (for example via PyPI packages) depends on those libraries; behavior may differ from non-FIPS images for
algorithms that are not allowed under FIPS policy.To migrate your application to a Docker Hardened Image, you must update your Dockerfile. At minimum, you must update the base image in your existing Dockerfile to a Docker Hardened Image. This and a few other common changes are listed in the following table of migration notes.
| Item | Migration note |
|---|---|
| Base image | Replace your base images in your Dockerfile with a Docker Hardened Image. |
| Package management | Non-dev images, intended for runtime, don't contain package managers. Use package managers only in images with a dev tag. |
| Non-root user | By default non-dev images, intended for runtime, run as the nonroot user. Ensure that necessary files and directories are accessible to the nonroot user. |
| Multi-stage build | Utilize images with a dev tag for build stages and non-dev images for runtime. To ensure that your final image is as minimal as possible, you should use a multi-stage build. While intermediary stages will typically use images tagged as dev, your final runtime stage should use a non-dev image variant. |
| TLS certificates | Docker Hardened Images contain standard TLS certificates by default. There is no need to install TLS certificates. |
| Ports | Non-dev hardened images run as a nonroot user by default. As a result, applications in these images can't bind to privileged ports (below 1024) when running in Kubernetes or in Docker Engine versions older than 20.10. To avoid issues, configure your application to listen on port 1025 or higher inside the container. |
| Entry point | Docker Hardened Images may have different entry points than images such as Docker Official Images. Inspect entry points for Docker Hardened Images and update your Dockerfile if necessary. |
| No shell | By default, image variants intended for runtime don't contain a shell. Use dev images in build stages to run shell commands and then copy any necessary artifacts to the runtime stage. |
The following steps outline the general migration process.
Find hardened images for your app.
A hardened image may have several variants. Inspect the image tags and find the image variant that meets your needs.
Update the base image in your Dockerfile.
Update the base image in your application's Dockerfile to the hardened image you found in the previous step. For
framework images, this is typically going to be an image tagged as dev because it has the tools needed to install
packages and dependencies.
For multi-stage Dockerfiles, update the runtime image in your Dockerfile.
To ensure that your final image is as minimal as possible, you should use a multi-stage build. All stages in your
Dockerfile should use a hardened image. While intermediary stages will typically use images tagged as dev, your
final runtime stage should use a non-dev image variant.
Install additional packages
Docker Hardened Images contain minimal packages in order to reduce the potential attack surface. You may need to install additional packages in your Dockerfile. Inspect the image variants to identify which packages are already installed.
The following are common issues that you may encounter during migration.
The hardened images intended for runtime don't contain a shell nor any tools for debugging. The recommended method for debugging applications built with Docker Debug (https://docs.docker.com/reference/cli/docker/debug/) to attach to these containers. Docker Debug provides a shell, common debugging tools, and lets you install other tools in an ephemeral, writable layer that only exists during the debugging session.
By default image variants intended for runtime, run as the nonroot user. Ensure that necessary files and directories are accessible to the nonroot user. You may need to copy files to different directories or change permissions so your application running as the nonroot user can access them.
Non-dev hardened images run as a nonroot user by default. As a result, applications in these images can't bind to
privileged ports (below 1024) when running in Kubernetes or in Docker Engine versions older than 20.10. To avoid issues,
configure your application to listen on port 1025 or higher inside the container, even if you map it to a lower port on
the host. For example, docker run -p 80:8080 my-image will work because the port inside the container is 8080, and
docker run -p 80:81 my-image won't work because the port inside the container is 81.
By default, image variants intended for runtime don't contain a shell. Use dev images in build stages to run shell
commands and then copy any necessary artifacts into the runtime stage. In addition, use Docker Debug to debug containers
with no shell.
Docker Hardened Images may have different entry points than images such as Docker Official Images. Use docker inspect
to inspect entry points for Docker Hardened Images and update your Dockerfile if necessary.