在北京超算V100-32G显卡上跑WRF是一种怎样的体验？

1WRF on GPU？YES!

WRF是在美国国家大气研究中心(NCAR)开发的天气研究和预报模型。它被162个国家的36000多个注册用户广泛使用。WRF具有多个动态核心，支持大规模并行计算，且系统可扩展性很强。WRF 适用于从米到数千公里的广泛应用。它是气象学科中最为广泛使用的数值模式之一。

熟悉WRF模式的小伙伴都知道，WRF是在CPU上运行的。而随着高性能计算技术，特别是使用图形处理器（GPU）等硬件进行大规模并行计算的技术正在日渐成熟，并在人工智能等领域不断长足发展。

那么，既然WRF如此适合大规模并行计算，那么，我们可以在GPU上运行WRF，以达到减低成本、增加效率的效果吗？答案当然是YES。

2AceCast: WRF的GPU版本

    AceCast是美国TempoQuest (TQI)公司开发的软件产品。其前身为英伟达公司支持开发的WRF-G。它由GPU提供支持，可以加速WRF模型。AceCAST 是五年来一丝不苟的研究和开发的产物，它使得 WRF 用户能够利用 GPU 硬件与传统 CPU 计算的高度并行性来保证优化性能。AceCAST 包含了大量重构的通用 WRF 物理、动力学模块和namelist，它使用了NVIDIA CUDA 或 OpenACC GPU编程技术，允许广大用户几乎不需要改变任何配置，就能将AceCAST作为现有WRF的平替，并加以使用。

    GPU通过使用具有高速计算速度和非常高的内存带宽的多线程、多核处理器来实现异常的加速。通用超级计算、高并行性、高内存带宽、低成本和紧凑体积的综合特性使得基于 gpu 的系统成为由普通 CPU 集群组成的大规模并行处理机/计算机系统的一个有吸引力的替代品。基于 GPU 开发的WRF 是目前世界上最快、分辨率最高的天气预报模型。

    TempoQuest 通过利用GPU加速 WRF 模型解决了这个问题。与标准的计算方法相比，AceCAST 是目前世界上最快和分辨率最高的天气预报模型。通过 AceCAST 运行 WRF，使用户能够以更高的分辨率、更低的成本和更深的洞察力运行预测和研究模拟，并加速解决时间流程。AceCAST 的能力为气象学家和最终用户提供了 WRF 产品，这些产品能够提高对全球气象模型无法识别或在低分辨率情况下无法识别的局部天气现象的认识。计算性能的提高使得高分辨率的确定性和概率性预报成为可能，而且比在中央处理器上运行 WRF 成本更低，预报模拟加速度一致地将预报处理速度提高了5倍。

3AceCast实战测试

    为了准确评估比较WRF在GPU和CPU上的运行速度，我和北京超算团队一起，使用北京超算提供的8卡V100-32G显卡资源，对一个800×600×33，单层嵌套的个例进行模拟。分别进行了1卡、2卡、4卡和8卡四个实验，主要计算其运行效率。实验结果如下：

4对比：CPU上运行WRF的效率

5对比：性能和价格

在北京超算上，一张V100-32G显卡的价格约为5元/小时（价格根据用户合同和使用时间等因素不同，会有所波动，下同。）；一个CPU核心的价格约为0.1元/小时。由此可以估算出，这次试验的消费金额（单精度CPU节点以40%提速预计，未进行实地实验）：

6总结与讨论

由表格我们可以得出以下结论：

总结一句话：要是WRF规模不太大，直接上单卡3090！性价比最高，最省时间！

7参考文献

扫码免费领取2000核时或200元卡时计算资源！