从源码编译安装 AutoGluonGPU 版基于 uv Workspace 的一键式安装指南【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon本指南讲解如何在本地从 GitHub 仓库源码安装支持 GPUCUDA的 AutoGluon。AutoGluon 官方仓库当前推荐使用 uv 作为包管理器整个仓库是一个 uv workspace仅需一条uv sync --all-extras命令即可把全部 7 个autogluon.*子包以可编辑editable方式安装到同一虚拟环境中。读完本文你将掌握 GPU 源码安装的完整流程、uv sync各参数的语义、CPU-only 变体、单子模块安装以及源码开发与依赖同步的最佳实践。一、为什么从源码安装GPU 场景对于绝大多数用户官方推荐直接pip install autogluon见 docs/install-gpu-pip.md或python -m uv pip install autogluon见 docs/install-gpu-uv.md。但当你需要以下能力时应从源码安装运行尚未发布的最新代码master 分支体验新特性或验证某个 bug 修复参与开发需要在本地直接修改autogluon.*源码并立即生效可编辑安装复现与官方 CI 一致的依赖环境uv.lock已提交到仓库保证确定性解析。需要特别说明的是GPU 支持本身不要求从源码编译。AutoGluon 的深度学习后端依赖 PyTorch而官方源码安装流程默认解析到的就是 Linux 上的 CUDA 版 PyTorch 构建因此uv sync --all-extras在 Linux NVIDIA GPU 环境下即为完整的 GPU 安装。这一点从仓库 CI 镜像也可印证CI/batch/docker/Dockerfile.gpu 直接基于官方 PyTorch GPU 训练镜像构建AutoGluon 本身以纯 Python 包形式安装不涉及 CUDA 扩展的本地编译。二、官方 GPU 源码安装命令核心docs/install-gpu-source.md给出的完整流程只有三条命令pip install -U uv git clone https://github.com/autogluon/autogluon cd autogluon uv sync --all-extras分步拆解如下升级/安装 uvpip install -U uv。uv 是 Astral 出品的 Python 包管理器官方文档明确建议优先使用uv安装方式这也是官方实际基准测试与 CI 所采用的安装路径。克隆仓库git clone https://github.com/autogluon/autogluon。若当前环境无法访问上述地址也可以直接克隆本仓库镜像git clone https://gitcode.com/GitHub_Trending/au/autogluon。同步依赖cd autogluon uv sync --all-extras。这一步会在仓库根目录创建.venv/虚拟环境并按提交的uv.lock解析、安装全部依赖及 7 个autogluon.*子包common、core、features、tabular、timeseries、multimodal以及元包autogluon全部以可编辑模式挂载修改源码无需重装。安装后的 GPU 验证同步完成后可以用如下命令确认 GPU 环境是否就绪uv run python -c import torch; print(torch.cuda.is_available(), torch.cuda.get_device_name(0))输出True及 GPU 型号名称即表示 PyTorch 成功识别到 CUDA 设备。同时可验证 AutoGluon 本体可正常导入uv run python -c from autogluon.tabular import TabularPredictor; print(ok)uv run会在不激活虚拟环境的情况下自动使用.venv/中的解释器执行命令你也可以先source .venv/bin/activateWindows 下为.venv\Scripts\activate再直接运行python。三、前置条件从源码安装前请确认以下环境详见 docs/install-from-source.md 与 docs/install.md组件要求Python3.10、3.11、3.12 或 3.13操作系统Linux / macOS / WindowsGPU 安装需 Linux NVIDIA 驱动 CUDA 环境git克隆仓库所需uv可用pip install -U uv或官方安装脚本curl -LsSf https://astral.sh/uv/install.sh \| sh官方文档特别提示AutoGluon 库已预装在所有 Amazon SageMaker Distribution 版本中SageMaker Studio 用户无需额外安装。四、uv sync参数深度解析docs/install-gpu-source.md中uv sync --all-extras是核心命令理解其参数语义有助于在不同场景下正确选用命令行为uv sync无参数仅安装基础依赖skeleton 安装不含可选模型uv sync --all-extras安装所有子包的全部可选依赖等价于pip install autogluon加上所有子模块 extras 的组合uv sync --frozen --all-extras严格按已提交的uv.lock原样安装不重新解析用于精确复现锁定环境仓库根目录的 pyproject.toml 中[tool.uv.workspace]声明了全部 7 个 workspace 成员[tool.uv.sources]则把autogluon.common、autogluon.core等兄弟包指向本地 workspace 成员而非 PyPI这正是uv sync能一次性可编辑安装整个 monorepo 的机制所在。4.1 GPU 与 CPU-only 的切换LinuxGPU/默认默认解析即拉取 CUDA 版 PyTorch也就是本指南所讲的 GPU 安装路径。CPU-only若机器无 GPU需显式指定 PyTorch CPU 索引uv sync --all-extras --index https://download.pytorch.org/whl/cpu这是docs/install-cpu-source.mdinstall-cpu-source.md给出的对应 CPU 版本与本篇 GPU 版本的区别仅在于多出的--index参数。macOSPyTorch 默认即为 CPU-only 构建直接uv sync --all-extras即可。4.2 指定 Python 解释器可通过uv sync -p 3.12选择特定解释器版本或使用uv python pin 3.12固定项目 Python 版本。五、只安装单个子模块并非所有场景都需要完整全家桶。uv sync --package允许只同步某个子包其兄弟autogluon.*依赖会从 workspace 自动解析# 只装 tabular且仅带 LightGBM 与 CatBoost 两个模型后端 uv sync --package autogluon.tabular --extra lightgbm --extra catboost # tabular 完整模型集等价于 pip install autogluon.tabular[all] uv sync --package autogluon.tabular --extra all其中 tabular 的allextras 包含lightgbm, catboost, xgboost, fastai, tabm, mitra, ray另有附加 extrastabicl, tabpfn, tabdpt, realmlp, interpret, imodels, skex, skl2onnx, tabpfnmix以及基准测试捆绑包tabarena。六、不克隆仓库的直接安装方式PEP 508如果只想临时安装某个未发布版本、而不想克隆整个 workspace可以使用 PEP 508 直接引用语法配合#subdirectory指向对应子包目录# 元包完整全家桶master 最新版 pip install autogluon githttps://github.com/autogluon/autogluon.git#subdirectoryautogluon # 单个子模块并带 extras pip install autogluon.tabular[all] githttps://github.com/autogluon/autogluon.git#subdirectorytabular # 固定分支 / tag / commit用 ref pip install autogluon.tabular githttps://github.com/autogluon/autogluon.gitmaster#subdirectorytabularuv pip install 同上亦可。需要注意的是此方式只构建指定子目录其兄弟包如autogluon.core、autogluon.features会从 PyPI 解析到对应已发布的固定构建版本而非 git 源码若想所有autogluon.*包都来自同一 git 源应使用第二节的uv sync克隆流程。七、源码开发工作流从源码安装的另一个核心用途是开发与测试。docs/install-from-source.md给出了完整的开发命令uv run pytest tabular/tests # 运行某个包的测试 uv run pre-commit install # 一次性启用格式化 hooks uv run pre-commit run --all-files # 像 CI 一样进行 lint/formatruffuv run pre-commit中使用的 ruff 配置定义在根目录 pyproject.toml 的[tool.ruff]段行宽 119lint.select显式指定为E4/E7/E9/F规则集。仓库各子包tabular/、timeseries/、multimodal/等都带有完整的tests/目录例如tabular/tests/unittests/下按模型lightgbm、xgboost、catboost 等划分的单元测试可直接用上述命令运行。八、保持与上游同步由于uv.lock已提交到仓库安装具有可复现性。每次拉取更新后重新执行uv sync --all-extras若你修改了某个包的依赖声明例如 core/_setup_utils.py 中的版本上限需要刷新并提交 lockfile——CI 会运行uv lock --checklockfile 过期会导致构建失败uv lock根目录pyproject.toml中通过[tool.uv]的exclude-newer 2026-08-01将依赖解析固定到某个日期之前因此新发布的依赖不会意外改变 lock 文件或触发 CI 报错直到你主动更新该日期并刷新uv lock。九、备选方案full_install.sh脚本对于不想接触 uv workspace 细节的场景仓库还提供了./full_install.shfull_install.sh。该脚本作为 pip 风格的兜底方案自动探测uv优先$UV环境变量其次 PATH 上的uv再次python -m uv都没有则自动通过 pip 安装然后对common、core、features、tabular、multimodal、timeseries、autogluon七个包逐一执行可编辑安装common/[tests]、core/[all,tests]、tabular/[all,tests]等。在 Colab 环境中脚本会自动强制非可编辑安装也可用--non-editable参数手动指定。此外仓库还提供了 full_install.bat 供 Windows 用户使用。十、安装注意事项与故障排查虚拟环境隔离uv sync会在仓库根目录创建.venv/与其他 Python 环境隔离避免污染系统 Python。GPU 驱动源码安装只解决 Python 依赖层CUDA 驱动与 PyTorch 的 CUDA 运行库仍需系统层面就绪可用nvidia-smi确认驱动可见。安装后异常官方文档建议在安装出现问题时先重启 notebook runtime 以清理内存状态若问题依旧可在仓库 issue 区反馈。nightly 构建若只想提前体验未发布修复而不从源码安装可执行python -m uv pip install --pre autogluon安装 nightly 构建但官方提示其未经稳定版同等严格的测试。小结从源码安装 AutoGluonGPU 版的本质是借助仓库的 uv workspace 结构用uv sync --all-extras一条命令完成 7 个子包的可编辑安装与 CUDA 版 PyTorch 的依赖解析。本文覆盖了官方docs/install-gpu-source.md的三条核心命令并延伸讲解了参数语义、CPU-only 变体、单子模块安装、PEP 508 直接引用、开发工作流与 lockfile 同步机制可作为源码级安装与开发环境的完整参考。【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考