clarknova9/mlflow

MLflow镜像（含PostgreSQL和MySQL客户端库）

镜像概述

本镜像基于MLflow构建，预安装PostgreSQL和MySQL的Python客户端库，旨在简化MLflow与这两种关系型数据库的集成。通过该镜像，用户可直接配置MLflow使用PostgreSQL或MySQL作为后端存储，实现机器学习实验数据、模型元数据的持久化管理。

核心功能与特性

• 完整MLflow支持：基于官方MLflow镜像构建，保留MLflow核心功能（实验跟踪、模型版本管理、模型部署等）
• 数据库客户端集成：预安装PostgreSQL客户端（psycopg2-binary）和MySQL客户端（mysql-connector-python）
• 无缝数据库连接：支持将MLflow后端存储（backend store）配置为PostgreSQL或MySQL数据库
• 环境兼容性：保持与官方MLflow镜像的API一致性，可直接替换标准MLflow镜像使用

使用场景与适用范围

• 数据科学团队需要将MLflow实验数据存储到PostgreSQL或MySQL数据库的场景
• Docker化部署MLflow服务，需与现有PostgreSQL/MySQL数据库集成的场景
• 多环境（开发/测试/生产）中保持MLflow配置一致性的需求
• 快速搭建包含数据库支持的MLflow服务原型

使用方法与配置说明

基本使用（Docker Run）

连接PostgreSQL数据库

bash
docker run -d -p 5000:5000 \
  -e MLFLOW_TRACKING_URI=postgresql://<user>:<password>@<postgres-host>:5432/<db-name> \
  <image-name> mlflow server \
  --backend-store-uri ${MLFLOW_TRACKING_URI} \
  --default-artifact-root ./artifacts \
  --host 0.0.0.0

连接MySQL数据库

bash
docker run -d -p 5000:5000 \
  -e MLFLOW_TRACKING_URI=mysql+pymysql://<user>:<password>@<mysql-host>:3306/<db-name> \
  <image-name> mlflow server \
  --backend-store-uri ${MLFLOW_TRACKING_URI} \
  --default-artifact-root ./artifacts \
  --host 0.0.0.0

Docker Compose配置示例

yaml
version: '3.8'
services:
  mlflow:
    image: <image-name>
    ports:
      - "5000:5000"
    environment:
      - MLFLOW_TRACKING_URI=postgresql://mlflow_user:mlflow_password@postgres:5432/mlflow_db
    depends_on:
      - postgres
    command: >
      mlflow server 
      --backend-store-uri ${MLFLOW_TRACKING_URI} 
      --default-artifact-root /mlflow/artifacts 
      --host 0.0.0.0
    volumes:
      - mlflow_artifacts:/mlflow/artifacts

  postgres:
    image: postgres:15
    environment:
      - POSTGRES_USER=mlflow_user
      - POSTGRES_PASSWORD=mlflow_password
      - POSTGRES_DB=mlflow_db
    volumes:
      - postgres_data:/var/lib/postgresql/data

volumes:
  mlflow_artifacts:
  postgres_data:

关键配置参数说明

注意事项

1. 确保数据库服务（PostgreSQL/MySQL）已启动并可被MLflow容器网络访问
2. 首次启动时，MLflow会自动在数据库中创建所需表结构，无需手动初始化
3. 生产环境建议使用持久化卷（volume）存储工件数据，避免容器重启导致数据丢失
4. 数据库连接字符串需包含正确的用户名、密码、主机地址、端口和数据库名