用途:Aurora 是微软面向地球系统的基础模型,本技能覆盖两条路径——推理与微调。推理用于区域天气/气候预报(以挪威示例为起点,含前端可视化);微调用于自定义变量、区域或预报时效,适合有明确应用需求的场景。\n\n主要能力:本地及 Azure 辅助的 Aurora 推理;在基础检查点上端到端微调;ERA5/CDS 数据下载、校验与批量构建;评估、检查点选择、训练不稳定排查。\n\n使用方式:先确定路径。探索/首次体验走推理路径:docs/about-aurora.md → docs/quick-start-inference.md,区域适配看 docs/customization-guide.md;真实应用走微调路径:先运行 initialization/initialize_starter_code.py(默认加 --skip-tests),再在 aurora-finetune/ 下执行 uv run pytest tests/test_training.py::test_finetuning_2t_var_pretrained --maxfail=1 验证,然后按 docs/quick-start-finetune.md、docs/finetuning-guide.md 推进。微调可用 uv run tensorboard --logdir tb_logs/finetuning 查看损失曲线(http://localhost:6006)。涉及新区域或 ERA5 下载时先执行 uv pip install cdsapi。\n\n环境要求:Windows 需在 WSL2(建议 Ubuntu)中运行,NVIDIA 驱动装在 Windows 主机;Python 3.11+ 与 uv;torch、lightning、numpy、xarray、microsoft-aurora>=1.5.2;挪威示例前端需 Node.js 18+ 与 npm;约 6 GB 磁盘空间(检查点约 5 GB)。推理可无 GPU 冒烟运行,认真训练强烈建议 A100 级 GPU。\n\n限制与注意:原生 Windows Python/PowerShell 不受支持;npm install 与 npm run dev 必须在 assets/inference/norway-example/frontend/ 下运行,仓库根目录会报错;推理快速开始中前端开发服务器需独占终端;CDS 凭据需本地编辑 .env,不得显示或提交 API 密钥;错误统一参考 docs/troubleshooting.md。更多细节(如完整硬件矩阵)暂未在资料中展开。
msresearch-aurora
写作与研究
更新于 10-16 06:47