ARTICLE DETAIL

资讯详情

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

3 分钟跑通 AutoGluon AutoML 环境:从 CPU 到 GPU 的完整部署指南

3 分钟跑通 AutoGluon AutoML 环境:从 CPU 到 GPU 的完整部署指南 3 分钟跑通 AutoGluon AutoML 环境从 CPU 到 GPU 的完整部署指南【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon刚装完 AutoGluon 那个「3 行代码完成训练与预测」的 AutoML 框架对着终端不知道该敲哪条命令本文 2 分钟带你从零基础到装好并验证通过不用记任何依赖关系按表格选路、照抄命令即可。30 秒决策你的场景走哪条路你的场景走哪条路只想跑通表格数据预测CPU 版 pip 安装下文最快路径有 NVIDIA GPU要训练多模态 / 深度学习模型GPU 版 pip 安装只少一行参数只需要时间序列或表格等单一能力单独装子模块依赖更轻要改源码、提 PR源码安装前提只有一条Python 版本在 3.10–3.13 之间。先跑python --version确认超范围就换个环境再装。最快上手路径两条命令装好pip install -U pip # CPU 机器保留 --extra-index-urlNVIDIA GPU 机器删掉该参数默认装带 CUDA 的 torch pip install autogluon --extra-index-url https://download.pytorch.org/whl/cpu装完用这段最小示例确认框架可用合成数据几秒就能拟合完import pandas as pd from autogluon.tabular import TabularPredictor df pd.DataFrame({a: range(100), b: [i % 2 for i in range(100)]}) predictor TabularPredictor(labelb).fit(df) # label 是目标列 print(predictor.fit_summary())能打印出模型评分表说明安装链路通了。CPU 分支差异按环境补一条前置命令结论CPU 机器唯一要注意的就是 pip 源参数别拉下几个 GB 的 CUDA 版 torch。Mac 用户先装 LightGBM 依赖的多线程库否则 import 时直接报错brew install libompApple SiliconM1/M2建议走 conda 路线conda-forge 会装 arm64 构建比 pip 省事。官方基准测试以 uv 安装为准追求安装速度可以换pip install -U uv后把 pip 命令换成python -m uv pip install参数不变。GPU 分支差异只有一行的区别结论NVIDIA GPU 机器上裸装即可默认 PyTorch wheel 自带 CUDA。pip install -U pip pip install autogluonGPU 服务器更常见的是 conda 路线一次配好 Python 与 CUDA 版 torchconda create -n ag python3.11 -y conda activate ag conda install -c conda-forge autogluon pytorch*cuda*适用条件驱动版本要能覆盖所装 torch 的 CUDA 版本。不确定是否装对直接看下文验收清单。子模块与源码分支只装你要的能力AutoGluon 按 tabular / multimodal / timeseries 模块化装全量不如按需选pip install autogluon.tabular # 默认是精简骨架版 pip install autogluon.tabular[all] # 与全量安装等价的表格能力 pip install autogluon.timeseries # 只装时间序列要二次开发就装源码仓库地址固定用下面这条git clone https://gitcode.com/GitHub_Trending/au/autogluon cd autogluon ./full_install.sh⚠️ 坑与解法症状 → 一条命令修复Mac 报libomp.dylib not foundbrew install libomp安装时报 Python 版本不支持conda create -n ag python3.11 -y后重新激活再装torch 提示 CUDA 版本不匹配pip install autogluon --extra-index-url https://download.pytorch.org/whl/cu121conda 依赖反复冲突官方明确建议改用 pip / uv 安装conda 的依赖差异可能影响稳定性✅ 验收清单三条命令确认装对了第一步确认框架可导入且版本正常当前版本 1.6.2关键标识是无 traceback 地打印出版本号python -c import autogluon; print(autogluon.__version__)GPU 机器再确认 CUDA 可用关键标识是输出里的Truepython -c import torch; print(torch.version.cuda, torch.cuda.is_available())最后跑一遍最快路径里的 3 行 fit 示例fit_summary()正常打印即算完成验收。 资源地图安装总入口uv / conda / 源码等全部分支docs/install.md子模块与可选依赖说明docs/install-modules.md各能力可运行示例examples/【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表