使用NVIDIA Earth2Studio构建定制批量集成天气预报

本文介绍了如何利用NVIDIA的Earth2Studio工具包构建一个集成天气预报工作流。通过安装必要的组件并保留Colab现有的CUDA启用PyTorch环境，加载FCN预测模型，并从GFS获取大气初始条件。文中还实现了将10米风分量转换为涡轮机容量因子的定制诊断方法以及应用物理上合适的噪声幅度到不同大气变量中的可变尺度扰动系统。利用Earth2Studio提供的低级迭代器、坐标映射、批量处理和Zarr API，构建了自定义集成执行管道，并将预报和诊断字段写入坐标感知数据存储中，最后通过纬度加权RMSE、公平CRPS等指标验证预报结果。