ARTICLE DETAIL

资讯详情

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

AutoGluon 自动化机器学习框架安装指南:覆盖 Linux、Mac、Windows 的三套命令

AutoGluon 自动化机器学习框架安装指南:覆盖 Linux、Mac、Windows 的三套命令 AutoGluon 自动化机器学习框架安装指南覆盖 Linux、Mac、Windows 的三套命令【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluonAutoGluon 是一个覆盖表格数据、图像、文本与时间序列的自动化机器学习AutoML框架。本文在 Python 3.9-3.12 的前提下给出安装环境要求、默认安装命令、CPU 与 GPU 的参数差异、按需安装模块以及最小验证代码供你按步骤在自己的机器上复现 AutoGluon 安装。环境准备核对开始 AutoGluon 安装之前先核对以下三项任一项不满足时先调整环境再执行安装命令项要求说明Python 版本3.9-3.12官方声明支持安装的版本区间支持操作系统Linux、MacOS、Windows三套系统各自有对应命令见后文速查表包管理器pip 推荐conda 安装可能因依赖差异导致性能不稳定完整安装文档入口为 docs/install.md与本文命令一一对应本文只整理 pip 主流程这是官方推荐的、兼容性与性能表现最佳的安装方式。一条命令安装CPU 与 GPU 参数差异无论哪种系统第一步都是升级两个构建工具避免后续安装过程因工具版本过低而报错pip install -U pip pip install -U setuptools wheel随后执行安装命令。CPU 场景是默认推荐形态--extra-index-url参数会引导 pip 从 PyTorch CPU 专用索引拉取 CPU 版 PyTorch 轮子pip install autogluon --extra-index-url https://download.pytorch.org/whl/cpuGPU 场景则去掉该参数直接从默认索引安装以便自动匹配 CUDA 加速所需的依赖pip install autogluon两种形态的完整步骤分别见 docs/install-cpu-pip.md 与 docs/install-gpu-pip.md。平台差异速查三个系统共用升级工具 安装的主流程差异集中在前置步骤与额外依赖系统基础命令差异额外依赖Linux无前置步骤直接执行上文两套命令无Mac与 Linux 相同的 CPU 安装命令需先执行brew install libompWindows创建环境后执行相同命令需先用 Anaconda 创建 Python 3.11 环境最容易踩的两处差异Mac 版必须先通过 Homebrew 安装 libomp 库否则安装或运行阶段会缺少依赖Windows 建议先执行conda create -n myenv python3.11 -y与conda activate myenv建立独立环境再运行 pip 安装命令。对应文档见 docs/install-mac-cpu.md 与 docs/install-windows-cpu.md。按需安装模块AutoGluon 采用模块化设计你可以只安装所需的子模块从而减少资源占用模块功能安装命令autogluon.tabular表格数据 AutoMLpip install autogluon.tabular[all]autogluon.multimodal图像/文本/多模态任务pip install autogluon.multimodalautogluon.timeseries时间序列预测pip install autogluon.timeseries只跑表格数据场景时最轻量的组合是显式指定所需的两类模型pip install autogluon.tabular[lightgbm,catboost]全部模块清单及其可选依赖见 docs/install-modules.md。验证安装成功安装完成后编写以下 3 行最小验证代码这是标准表格场景加载示例数据集并限时 60 秒训练一个模型from autogluon.tabular import TabularDataset, TabularPredictor train_data TabularDataset(https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv) predictor TabularPredictor(labelclass).fit(train_data, time_limit60)脚本输出模型训练结果且无报错退出即表示安装成功完整可运行示例见 examples/tabular/example_simple_tabular.py。常见问题速查现象可能原因解决办法Windows 安装失败未安装 Anaconda或命令提示符未以管理员权限运行按 docs/install-windows-generic.md 安装 Anaconda并使用管理员权限运行命令提示符Mac M1/M2 芯片无法正常安装pip 安装路径在 Apple Silicon 上存在兼容问题改用conda install -c conda-forge autogluonGPU 版本出现 CUDA 兼容性报错CUDA 版本与 PyTorch 不兼容确认 CUDA 版本与 PyTorch 兼容参考 docs/install-gpu-pip.md后续路径官方教程docs/tutorials/速查表docs/cheatsheet.md示例代码examples/其他安装问题可在项目仓库的 issues 页面提交【免费下载链接】autogluonFast and Accurate ML in 3 Lines of Code项目地址: https://gitcode.com/GitHub_Trending/au/autogluon创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表