evk02/mlflow

dockerhub-mlflow

MLFlow的Docker镜像，适用于DockerHub和Helm图表部署。

您可以在DockerHub上找到此镜像：https://hub.docker.com/r/evk02/mlflow

注意：从v1.29.1版本开始，该镜像和推荐的Helm图表支持"场景5：启用MLflow Tracking Server并支持代理化制品存储访问"，详见[***]

Helm图表

推荐将此Docker镜像与Helm图表配合使用：https://github.com/zoltan-fedor/helm-mlflow

使用Helm图表安装到Kubernetes

前置要求

• MySQL数据库（需创建MLFlow专用用户）
• S3或S3兼容存储（如MinIO）

配置values.yaml文件

yaml
# MLFlow Helm图表自定义配置
# 完整配置选项详见https://github.com/cetic/helm-mlflow
image:
  repository: evk02/mlflow
  tag: 1.23.1
  pullPolicy: IfNotPresent

# MySQL数据库连接配置
db:
  default:
    enabled: false
  type: mysql  # 数据库类型，可选mysql或postgresql
  host: mysql_host  # MySQL主机地址
  port: 3306  # MySQL端口
  user: mlflow  # 数据库用户名
  password: MyPassword  # 数据库密码
  database: mlflow  # 数据库名称

# 制品存储路径
artifacts_destination: "s3://datalake/mlflow/artifacts"

# MLFlow UI超时配置
gunicorn_opts: "-t360"  # 超时时间360秒

# MinIO配置（如使用S3可忽略）
minio:
  enabled: false
  # url: http://minio-service.kubeflow.svc.cluster.local:9000
  # accesskey: minio
  # secretkey: MySecretKey

# 服务配置
service:
  type: ClusterIP  # 仅集群内访问
  port: 80

部署命令

bash
# 添加Helm仓库
$ helm repo add cetic https://cetic.github.io/helm-charts
# 更新仓库索引
$ helm repo update
# 安装MLFlow（指定命名空间mlflow）
$ helm install -f values.yaml mlflow cetic/mlflow -n mlflow

使用Helm图表和Terraform安装到Kubernetes

前置要求

• 同Helm部署要求（MySQL和S3/MinIO）
• 已创建上述values.yaml配置文件

Terraform配置示例

hcl
resource "helm_release" "mlflow" {
  name       = "mlflow"
  namespace  = "mlflow"
  create_namespace = true  # 自动创建命名空间

  repository       = "https://cetic.github.io/helm-charts"
  chart            = "mlflow"
  # version          = "1.5.1"  # 可选，指定Helm图表版本

  # 镜像配置
  set {
    name  = "image.repository"
    value = "evk02/mlflow"  # Docker镜像仓库
  }
  set {
    name  = "image.tag"
    value = "1.23.1"  # MLFlow应用版本
  }

  # 引用外部values配置文件
  values = [
    file("${path.module}/values.yaml")
  ]
}

升级说明

当版本升级后MLFlow容器出现错误时，通常是由于数据库schema需要更新。执行以下命令升级数据库：

bash
$ kubectl run -n mlflow -i --tty --rm mlflow-upgrade \
  --image=evk02/mlflow:1.29.2 --restart=Never \
  -- sh -c 'mlflow db upgrade [db uri]'

注：将[db uri]替换为实际数据库连接URI，格式为mysql+pymysql://user:password@host:port/database