用途:帮助用户按自身用例正确安装 Earth2Studio 及其可选模型依赖,覆盖包安装、可选 extras 选择、环境变量配置与安装验证。不负责编写推理或训练代码、组合工作流、超出 data extra 的数据源设置、模型检查点下载及与缺失依赖无关的运行时错误。
前置条件:Python 3.10+(推荐 3.13);GPU extras 需支持 CUDA 的 GPU 与兼容驱动;推荐 uv 或 pip;需联网(包来自 PyPI 和 GitHub)。
核心原则:安装命令、版本标签与 extras 名称随版本变化,应先获取实时安装文档 https://nvidia.github.io/earth2studio/userguide/about/install.html,解析当前版本标签、可用 extras 与特殊构建说明。严禁代用户安装、升级或修改包:不得运行 pip install、uv pip install、uv add、uv sync、conda install、apt install 等;应给出确切命令并请用户自行执行,说明原因并等待确认。
流程:先获取实时文档;再了解用户环境(包管理器、项目上下文、Python 版本,最多 3 个问题);给出基础安装命令,uv 使用 git 源、pip 从 PyPI 安装且部分 extras 需手动预装;安装后验证 import earth2studio 与 earth2studio.version;按用例(预报、诊断、数据同化、子模块 data/perturbation/statistics)选择 extras,uv 可用 --extra all;随后配置环境变量(EARTH2STUDIO_CACHE、EARTH2STUDIO_DATA_CACHE、EARTH2STUDIO_MODEL_CACHE、EARTH2STUDIO_PACKAGE_TIMEOUT),仅在相关时提出。
注意事项:flash-attention、natten、torch-harmonics 的 CUDA 扩展构建可能需 10–30+ 分钟;部分模型需 --no-build-isolation 或预装 earth2grid、torch-harmonics、makani;数据同化模型需 CuPy 与 cuDF(CUDA 12)。常见问题包括 PyTorch/CUDA 不匹配、flash-attention 构建失败、ONNX Runtime GPU 版本、ecCodes 缺失(Debian/Ubuntu 可 sudo apt-get install libeccodes-dev,或 conda install -c conda-forge eccodes)、缺少 Python.h(sudo apt-get install python3-dev)。故障排查参见官方 troubleshooting 与 FAQ 页面。