trivadis/mlflow-server

docker-mlflow-server 镜像文档

镜像概述

docker-mlflow-server是用于部署MLflow服务器的Docker镜像。MLflow是一个开源平台，用于管理机器学习全生命周期，包括实验跟踪、模型管理、项目打包和模型部署。该镜像提供了便捷的MLflow server部署方式，无需手动配置依赖环境，可快速搭建机器学习实验管理服务。

核心功能与特性

• 实验跟踪：记录和比较机器学习实验的参数、指标和输出工件（如模型文件、数据）
• 模型管理：提供模型版本控制、存储和元数据管理功能
• 标准化部署：支持将模型部署为REST API服务
• 轻量级部署：预配置MLflow server运行环境，简化部署流程
• 灵活配置：支持自定义存储后端（本地文件系统、云存储等）和数据库后端

使用场景

• 数据科学家进行机器学习实验跟踪与结果比较
• 机器学习团队协作管理模型版本和实验记录
• 快速搭建临时或生产环境的MLflow服务
• 集成到CI/CD流程中实现模型自动化部署

使用方法

基本运行

启动默认配置的MLflow服务器，使用本地文件系统存储实验数据：

bash
docker run -d -p 5000:5000 --name mlflow-server mlflow/server

访问 http://localhost:5000 即可打开MLflow UI界面。

配置持久化存储

将实验数据存储到主机目录，实现数据持久化：

bash
docker run -d -p 5000:5000 \
  --name mlflow-server \
  -v /path/to/local/mlruns:/mlflow/mlruns \
  mlflow/server

配置数据库后端

使用数据库（如PostgreSQL）存储实验元数据：

bash
docker run -d -p 5000:5000 \
  --name mlflow-server \
  -e MLFLOW_TRACKING_URI=postgresql://user:password@db-host:5432/mlflow_db \
  -v /path/to/mlruns:/mlflow/mlruns \
  mlflow/server

docker-compose配置示例

创建docker-compose.yml文件：

yaml
version: '3'
services:
  mlflow-server:
    image: mlflow/server
    ports:
      - "5000:5000"
    volumes:
      - ./mlruns:/mlflow/mlruns
    environment:
      - MLFLOW_TRACKING_URI=/mlflow/mlruns
    restart: always

启动服务：

bash
docker-compose up -d

环境变量配置