ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

AutoGluon 自动化机器学习安装指南:3 步跑通表格与多模态建模

AutoGluon 自动化机器学习安装指南:3 步跑通表格与多模态建模 AutoGluon 自动化机器学习安装指南3 步跑通表格与多模态建模【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluonAutoGluon 是面向表格、图像、文本与时间序列数据的自动化机器学习AutoML框架用几行代码即可完成数据预处理、模型选择与集成训练。按本文完成环境确认与安装后你将能独立部署 AutoGluon、验证安装结果并为自己的数据任务选择合适的功能模块。开始前的3项确认确认项要求语言版本Python 3.10 / 3.11 / 3.12 / 3.13操作系统Linux、macOS、Windows 均受支持CPU 与 GPU 差异CPU 版在安装命令中通过--extra-index-url指定 PyTorch CPU 索引源GPU 版使用默认源直接安装且需要本地驱动满足对应 PyTorch 的 CUDA 要求。macOS 暂不支持 GPU 加速如需 GPU 请改用 Linux 或 Windows官方推荐通过pip或uv安装其中uv是官方持续进行基准测试与验证的安装方式conda 安装因依赖细节差异可能影响性能与稳定性遇到问题时应优先回退到pip/uv方案。安装主路径核心命令CPU 版如下GPU 环境直接去掉--extra-index-url参数即可pip install -U pip pip install -U setuptools wheel pip install autogluon --extra-index-url https://download.pytorch.org/whl/cpu各平台的差异点macOS安装前先用 Homebrew 执行brew install libomp因为依赖 LightGBM 的多线程运行需要 libomp 运行库Windows建议使用 Anaconda 管理环境在 Anaconda Prompt 中执行conda create -n myenv python3.11 -y conda activate myenv随后按上方核心命令继续安装。Linux按核心命令直接执行无额外前置步骤Apple SiliconM 系列芯片可改用 conda 安装路径conda-forge 会按机器能力自动选择对应版本。装完先验证运行以下最小示例它会下载一个小型数据集并训练一个表格分类模型from autogluon.tabular import TabularDataset, TabularPredictor data TabularDataset(https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv).head(500) predictor TabularPredictor(labelclass).fit(data, time_limit60) print(predictor.leaderboard(silentTrue))成功运行时终端会先打印训练过程日志最后输出一个 leaderboard 表格列出各模型名称、得分与训练耗时且predictor.fit结束后会在当前目录生成模型保存路径。更多示例见 examples/tabular/。只装需要的模块AutoGluon 按数据形态拆分为独立子模块按需安装可以显著减少依赖数量与磁盘占用也避免无关组件引入版本冲突。例如只需表格功能且仅使用 LightGBM 与 CatBoost 两种模型时可执行pip install autogluon.tabular[lightgbm,catboost]模块功能安装命令autogluon.tabular表格数据 AutoMLTabularPredictorpip install autogluon.tabular[all]autogluon.multimodal图像、文本与多模态深度学习任务pip install autogluon.multimodalautogluon.timeseries时间序列预测TimeSeriesPredictorpip install autogluon.timeseries各模块的可选依赖清单见 官方模块安装文档。常见症状速查现象可能原因解决办法安装时报 Python 版本不兼容使用了官方支持的 3.10–3.13 之外的版本新建支持范围内的虚拟环境后重装Windows 可用 conda 指定python3.11macOS 上 LightGBM 训练时报 libomp 相关错误缺少 libomp 运行库执行brew install libomp后重启终端再运行想用 GPU 但 macOS 上无效果macOS 尚不支持 GPU 加速改用 Linux 或 Windows 环境GPU 环境启动时报 CUDA / 驱动不匹配CUDA 驱动版本与所装 PyTorch 版本不兼容按 官方 GPU 安装文档 核对 PyTorch 与 CUDA 的对应关系并重装安装过程反复失败或依赖冲突使用了 conda 安装且依赖解析出现差异换用官方推荐的pip或uv方式重新安装从安装到实践官方安装总入口docs/install.md分场景分平台教程目录docs/tutorials/可运行的示例代码examples/项目概览与特性说明README.md下一步建议先用time_limit参数跑通一个 60 秒的表格分类 demo熟悉TabularPredictor的 fit / predict / leaderboard 工作流后再按自己的数据形态切换到对应的功能模块。【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进