ARTICLE DETAIL

资讯详情

深耕郑州网站建设与运营推广的一线实战洞察。

AutoGluon 安装上手实操手册:从环境体检到验证通过,四步走完全流程

AutoGluon 安装上手实操手册:从环境体检到验证通过,四步走完全流程 AutoGluon 安装上手实操手册从环境体检到验证通过四步走完全流程【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon给 AutoGluon 这个 AutoML 框架配环境最容易翻车的是选错路线CPU 机器装成 GPU 版拉下几 GB 依赖或 CPU 版漏掉额外参数导致拉到的依赖跑不起来。本文按环境体检 → 选路线 → 安装 → 验证的顺序展开跟着做即可完成安装并验证。先把 AutoGluon 的定位讲清楚先搞清楚你装的是什么。AutoGluon 是一个开源 AutoML 框架把表格、时间序列、图像文本等多模态数据的机器学习流程自动化你给出数据、指定要预测的列它会在从经典树模型到基础模型的范围内自动挑选并组合出最适合你数据的那套模型三行代码就能拿到可用的预测结果。做表格预测、时间序列预测、图像或文本任务又希望不亲手调参也能拿到强基线的人就是它的目标用户。 给环境做个体检先核对四项这一步决定后面所有命令能不能跑通。打开终端对照下表逐项检查检查项要求不达标怎么办Python 版本3.10–3.13新建环境conda create -n myenv python3.11 -y操作系统Linux / MacOS / Windows 均支持无需处理包管理器官方推荐 pip 或 uv改用 pip/uvconda 的依赖差异已被官方点名可能影响性能与稳定性Mac 额外依赖libompLightGBM 多线程所需brew install libomp官方安装文档 docs/install.md 明确说uv 和 pip 是团队主动做基准测试的安装方式conda 安装可能存在细微的依赖差异。所以下面所有路线都默认走 pip除非你的环境只能用 conda。 按使用场景挑安装路线这一步把机器状态映射到一条命令上对号入座只执行其中一条。纯 CPU 机器一条命令装完 CPU 版多数读者是这个场景。这条命令先升级 pip 和两个打包工具再安装 AutoGluon--extra-index-url参数负责把 PyTorch 锁定到 CPU 版 wheel避免拉下不需要的 CUDA 依赖pip install -U pip pip install -U setuptools wheel pip install autogluon --extra-index-url https://download.pytorch.org/whl/cpu看到最后输出Successfully installed autogluon-...加一串依赖名、且没有红色报错即算成功。GPU 机器装 GPU 加速版只有当你有 NVIDIA GPU、并且确实想用 GPU 训练深度学习模型时才走这条路线。这条命令会默认安装带 CUDA 的 PyTorchpip install -U pip pip install -U setuptools wheel pip install autogluon注意装完要确认本机 CUDA 驱动与 PyTorch 版本匹配否则 GPU 用不起来处理办法见文末踩坑急救。细节参考 docs/install-gpu-pip.md。特殊环境Windows 与 M 系列芯片Windows官方推荐走 Anaconda。下面这条命令一次完成创建 3.11 环境、激活、安装 CPU 版在 Anaconda Prompt 中直接执行即可conda create -n myenv python3.11 -y conda activate myenv pip install -U pip pip install -U setuptools wheel pip install autogluon --extra-index-url https://download.pytorch.org/whl/cpuM1/M2Apple Silicon官方文档写明 Apple Silicon 通过 conda 路线支持conda-forge 会在机器支持时自动装 GPU 版可照 docs/install-conda-full.md 执行走 pip 路线也没问题但先执行brew install libomp。✂️ 按需裁剪只装你要用的模块如果你的任务只涉及某一类数据只装对应子模块依赖更少、装得更快模块适用任务安装命令autogluon.tabular表格分类/回归TabularPredictorpip install autogluon.tabular[all]autogluon.multimodal图像、文本、多模态MultiModalPredictorpip install autogluon.multimodalautogluon.timeseries时间序列预测TimeSeriesPredictorpip install autogluon.timeseries两个容易忽略的点单独安装autogluon.tabular得到的是骨架安装不含任何模型库加[all]才与默认pip install autogluon的表格模型集合一致如果只要个别模型还能进一步裁剪例如pip install autogluon.tabular[lightgbm,catboost]。全部模块与可选依赖清单见 docs/install-modules.md。✅ 跑最小示例验收安装最快的验收方式加载演示表格数据、限时 60 秒训练、打印模型排行榜from autogluon.tabular import TabularDataset, TabularPredictor train_data TabularDataset(https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv) train_data train_data.head(500) predictor TabularPredictor(labelclass).fit(train_data, time_limit60) predictor.leaderboard(print_onlyTrue)成功的样子很具体终端打印出一张 leaderboard 表列出各候选模型及其验证集得分且当前目录下生成了模型保存目录。能看到这张表就说明安装已经通过验收。完整训练 推理 评估版本参考示例 examples/tabular/example_simple_tabular.py。 踩坑急救四个高频问题各三句话症状Mac 上 import 到 LightGBM 时崩溃日志提示找不到libomp。→定位未安装 libomp 多线程库。→解法brew install libomp后重新运行。症状conda 装是能跑但行为或性能与别人 pip 装的不一致。→定位conda 安装存在细微的依赖差异。→解法改用 pip 或 uv 重装见 docs/install.md。症状Windows 上安装失败、报依赖解析错误。→定位不在干净的 Anaconda 环境里。→解法打开 Anaconda Prompt先创建 Python 3.11 环境再执行 pip 命令。症状GPU 版装完后torch.cuda.is_available()返回 False。→定位本机 CUDA 驱动与 PyTorch 版本不匹配。→解法核对 PyTorch 与 CUDA 的兼容矩阵参考 docs/install-gpu-pip.md。 下一步装完之后的四个入口装好并验收通过后按这个顺序深入完整安装文档含 uv、conda、源码安装、nightly 构建docs/install.md表格任务快速上手教程docs/tutorials/tabular/tabular-quick-start.ipynb功能速查表docs/cheatsheet.md可运行示例代码目录examples/【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表