wandb/launch-agent

镜像概述

本镜像为 Weights & Biases (wandb) 启动代理，基于 wandb 官方仓库构建，旨在协调和执行机器学习任务。作为 wandb 生态的一部分，该代理负责接收、调度和监控任务执行，同时与 wandb 的实验跟踪系统深度集成，确保任务执行过程中的指标、日志和结果可被实时跟踪与分析。

核心功能与特性

• 官方仓库同步：基于 wandb 官方仓库最新代码构建，确保功能与上游保持一致。
• 任务调度与执行：支持接收并执行来自 wandb 平台的机器学习任务，包括训练、评估等工作流。
• 实验跟踪集成：原生集成 wandb 的实验跟踪功能，自动记录任务执行过程中的指标、超参数和结果。
• 跨环境兼容性：可部署于本地环境、私有服务器或云平台，适应不同规模的机器学习基础设施。
• 轻量级设计：优化资源占用，适合在各类计算环境中高效运行。

使用场景

• 机器学习团队协作：团队成员通过 wandb 平台提交任务，由启动代理统一调度执行，提升协作效率。
• 自动化实验管理：实现机器学习实验的自动化执行与跟踪，减少手动操作成本。
• 分布式任务协调：在多节点或集群环境中，作为任务协调者分配计算资源并监控执行状态。
• CI/CD 集成：嵌入机器学习模型开发的 CI/CD 流程，实现模型训练与评估的自动化触发。

使用方法

前提条件

• 已注册 wandb 账号并获取 API 密钥（可在 wandb 设置页面 获取）。
• 目标环境已安装 Docker。

快速启动

使用以下命令启动 wandb 启动代理：

bash
docker run -d \
  --name wandb-launch-agent \
  -e WANDB_API_KEY="your_wandb_api_key" \
  -e WANDB_AGENT_NAME="my-agent" \
  wandb/launch-agent:latest

环境变量配置

启动代理支持通过环境变量进行自定义配置，常用参数如下：

私有部署配置

若使用私有 wandb 服务器，需指定 WANDB_BASE_URL：

bash
docker run -d \
  --name wandb-launch-agent \
  -e WANDB_API_KEY="your_private_api_key" \
  -e WANDB_BASE_URL="https://your-private-wandb-server.com" \
  -e WANDB_QUEUE="private-queue" \
  wandb/launch-agent:latest

容器日志查看

通过以下命令查看代理运行日志：

bash
docker logs -f wandb-launch-agent

停止与重启

bash
# 停止容器
docker stop wandb-launch-agent

# 重启容器
docker restart wandb-launch-agent

注意事项

• 确保 WANDB_API_KEY 保密，避免泄露账号权限。
• 私有部署时需确保代理可访问指定的 WANDB_BASE_URL 服务器。
• 代理版本需与 wandb 平台版本兼容，建议使用 latest 标签获取最新稳定版。