determined

GitHub
3.2k 371 较难 1 次阅读 3天前Apache-2.0开发框架数据工具其他
AI 解读 由 AI 自动生成,仅供参考

Determined 是一个开源的深度学习平台,旨在让机器学习模型的训练与管理变得更加简单高效。它完美兼容 PyTorch 和 TensorFlow 两大主流框架,帮助开发者轻松应对分布式训练、超参数调优、实验追踪及资源管理等复杂挑战。

在深度学习项目中,手动配置多卡并行训练往往门槛较高,云端的 GPU 资源成本也难以控制,同时大量的实验记录容易导致结果难以复现。Determined 通过一体化的解决方案解决了这些痛点:它能自动加速分布式训练以缩短等待时间,利用智能算法自动寻找最佳模型参数,并通过精细的资源调度降低云端算力开销。此外,其内置的实验追踪功能可完整记录代码快照与性能指标,确保研究过程可追溯、可复现。

这款工具特别适合机器学习工程师、数据科学家以及 AI 研究人员使用。无论是希望在本地快速搭建集群的原型开发者,还是需要在 AWS、GCP 或 Kubernetes 上管理大规模训练任务的企业团队,都能从中受益。

Determined 的独特之处在于其灵活的接入方式:用户既可以通过简单的 Python 类封装快速迁移现有代码,也能利用强大的命令行界面(CLI)和直观的 Web UI 进行全流程管控。只需简单的 YAML 配置文件,即可定义从资源分配到搜索策略的所有细节,让复杂的深度学习工作流变得井井有条。

使用场景

某电商推荐算法团队正在基于 PyTorch 训练大规模用户行为预测模型,急需在有限预算下快速迭代出最优参数组合。

没有 determined 时

  • 分布式训练需手动编写复杂的多卡通信代码,调试耗时且容易出错,导致模型收敛速度缓慢。
  • 超参数调优依赖人工试错或简陋脚本,无法智能搜索最佳组合,往往错过全局最优解。
  • GPU 资源闲置与争抢并存,缺乏统一调度机制,云厂商账单高昂且资源利用率低下。
  • 实验记录散落在本地日志或笔记中,难以复现特定版本的模型效果,团队协作效率极低。

使用 determined 后

  • 仅需通过 YAML 配置文件即可一键启动 8 卡分布式训练,determined 自动处理底层通信,训练速度提升数倍。
  • 内置自适应搜索算法(如 Adaptive ASHA)自动探索学习率等关键参数,以更少试验次数找到精度更高的模型。
  • 动态资源管理根据任务优先级自动分配和回收 GPU 插槽,显著降低云端算力成本,避免资源浪费。
  • Web UI 集中展示所有实验的损失曲线、代码快照及性能报告,团队成员可随时对比分析并无缝复现结果。

determined 将繁琐的基础设施运维转化为简单的配置工作,让算法工程师能专注于模型创新而非工程杂务。

运行环境要求

操作系统
  • 未说明
GPU

需要 GPU 以进行分布式训练和降低成本(具体型号、显存及 CUDA 版本未在 README 中明确说明,但提及支持 AWS/GCP 云 GPU)

内存

未说明

依赖
notes该工具是一个全功能的深度学习平台,支持通过 CLI 在本地、AWS、GCP、Kubernetes 或 Slurm/PBS 上部署集群。用户需将代码封装为特定类或使用 Core API 以兼容该平台。实验配置(如分布式训练槽位、超参数搜索策略)通过 YAML 文件管理。具体系统要求需参考官方文档中的集群部署指南。
python未说明
PyTorch
TensorFlow
determined (CLI/Library)
determined hero image

快速开始

Determined AI Logo

Determined 是一个一体化深度学习平台,兼容 PyTorch 和 TensorFlow。

它负责:

  • 分布式训练,以加快结果产出。
  • 超参数调优,以获得最佳模型。
  • 资源管理,以降低云 GPU 成本。
  • 实验跟踪,以便进行分析和复现。

Features gif

Determined 的工作原理

Determined 的主要组成部分包括 Python 库、命令行界面 (CLI) 和 Web UI。

Python 库

使用 Python 库可以使您现有的 PyTorch 或 TensorFlow 代码与 Determined 兼容。

您可以通过将代码组织到基于类的 API 中来实现这一点:

from determined.pytorch import PyTorchTrial

class YourExperiment(PyTorchTrial):
  def __init__(self, context):
    ...

或者通过 Core API 使用您需要的函数:

import determined as det

with det.core.init() as core_context:
    ...

命令行界面 (CLI)

您可以使用 CLI 来:

  • 在本地启动 Determined 集群:
det deploy local cluster-up
  • 在云服务上部署 Determined,例如 Amazon Web Services (AWS) 或 Google Cloud Platform (GCP):
det deploy aws up
  • 训练您的模型:
det experiment create gpt.yaml .

使用 YAML 文件配置从分布式训练到超参数调优的所有内容:

resources:
  slots_per_trial: 8
  priority: 1
hyperparameters:
  learning_rate:
    type: double
    minval: .0001
    maxval: 1.0
searcher:
  name: adaptive_asha
  metric: validation_loss
  smaller_is_better: true

Web UI

使用 Web UI 可以查看损失曲线、超参数图表、代码和配置快照、模型注册表、集群利用率、调试日志、性能剖析报告等。

Web UI

安装

安装 CLI:

pip install determined

然后使用 det deploy 在本地或 AWS、GCP 等云服务上启动 Determined 集群。

有关安装详情,请访问适用于您环境的集群部署指南:

示例

通过探索 examples 文件夹determined-examples 仓库 中的 30 多个示例,熟悉 Determined。

文档

社区

如果您需要帮助、想要提交 bug 报告,或者只是想及时了解 Determined 的最新动态,请加入 Determined 社区!

贡献

贡献者指南

许可证

Apache V2

版本历史

v0.38.12025/03/19
v0.38.1-ee2025/03/19
v0.38.0-ee2025/03/07
v0.38.02024/11/22
0.35.12024/11/09
0.37.02024/09/30
0.36.02024/08/23
0.35.02024/08/08
0.34.02024/06/28
0.33.02024/05/29
0.32.12024/05/10
0.32.02024/05/08
0.31.02024/04/17
0.30.02024/04/04
0.29.12024/03/18
0.29.02024/03/05
0.28.12024/02/20
0.28.02024/02/06
0.27.12024/01/24
0.27.02024/01/10

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|1周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

154.3k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

108.3k|★★☆☆☆|3天前
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|4天前
插件Agent图像

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|1周前
插件开发框架