kelvinsp/mlflow

Mlflow 镜像

镜像概述

本Docker镜像用于运行Mlflow服务器，提供机器学习实验全生命周期管理能力。默认支持多种主流数据库和对象存储服务，便于快速部署和集成到机器学习工作流中。

核心功能与特性

• 多数据库支持：原生兼容PostgreSQL和MySQL数据库，用于存储实验元数据
• 多存储支持：支持MinIO和Amazon S3对象存储服务，用于存储模型 artifacts
• 开箱即用：无需复杂配置即可快速启动Mlflow服务器
• 灵活集成：可与现有数据科学工具链无缝集成

使用场景与适用范围

• 机器学习实验跟踪：记录实验参数、指标、代码版本和模型 artifacts
• 模型版本管理：管理不同版本模型及其元数据，支持模型追溯
• 团队协作：多用户共享实验结果，促进协作开发
• 适用人群：数据科学家、机器学习工程师、AI研究人员

使用方法与配置说明

基本运行命令

bash
docker run -p 5000:5000 kelvins/mlflow

上述命令将启动Mlflow服务器，默认使用本地文件系统存储元数据和artifacts，可通过http://localhost:5000访问Web界面。

数据库配置

通过环境变量配置数据库连接：

示例（PostgreSQL）：

bash
docker run -p 5000:5000 \
  -e DB_TYPE=postgresql \
  -e DB_URI=postgresql://mlflow:mlflow@postgres:5432/mlflow \
  kelvins/mlflow

存储配置

通过环境变量配置artifacts存储位置：

示例（Amazon S3）：

bash
docker run -p 5000:5000 \
  -e DB_TYPE=postgresql \
  -e DB_URI=postgresql://mlflow:mlflow@postgres:5432/mlflow \
  -e ARTIFACT_ROOT=s3://my-mlflow-bucket/artifacts \
  -e AWS_ACCESS_KEY_ID=your-access-key \
  -e AWS_SECRET_ACCESS_KEY=your-secret-key \
  kelvins/mlflow

Docker Compose示例

yaml
version: '3'
services:
  mlflow:
    image: kelvins/mlflow
    ports:
      - "5000:5000"
    environment:
      - DB_TYPE=postgresql
      - DB_URI=postgresql://mlflow:mlflow@postgres:5432/mlflow
      - ARTIFACT_ROOT=minio://minio:9000/mlflow-artifacts
    depends_on:
      - postgres
      - minio

  postgres:
    image: postgres:13
    environment:
      - POSTGRES_USER=mlflow
      - POSTGRES_PASSWORD=mlflow
      - POSTGRES_DB=mlflow
    volumes:
      - postgres_data:/var/lib/postgresql/data

  minio:
    image: minio/minio
    environment:
      - MINIO_ROOT_USER=minio
      - MINIO_ROOT_PASSWORD=minio123
    command: server /data
    volumes:
      - minio_data:/data

volumes:
  postgres_data:
  minio_data:

更多信息

访问Github仓库获取完整文档和更新：https://github.com/kelvins/docker-images