这些年随着GPU算力的不断发展,不少机构在尝试利用GPU加速当前的数值天气预报模式。至少到目前为止还没看到官方发布的完整支持GPU版本的数值预报模式。
图:与NVIDIA合作的气象机构(来源网络)
今天看了眼控科技在国家气象中心做的“AI技术在数值模式中的应用探讨”报告。报告中提到了利用GPU加速数值模式,PPT里给出了WSM6等微物理参数化的GPU和CPU对比。那今天就给大家介绍一下当前已经发布的社区版GPU加速的WRF模式-WRFg。
WRFg是利用GPU加速的社区版中小尺度WRF模式,包含了完整的WRF动力核及一些微物理选项,可以利用GPU、OpenACC和CUDA加速WRF模式的运行。官方给出的性能对比是,GPU加速的WRF模式比CPU版本的速度提高了7倍。
官方网站给出的信息,加速后的WRF模式可以运行1km分辨率,但是没有指明多大范围。加速后的WRF模式可以同化更多的观测数据,以改善数值模式的初始场。
网站发布的是基于WRF V3.8.1的加速版本。以下是当前已经移植到GPU的一些子模块信息。
官方指出,WRFg可以部署在NVIDIA P100或V100的系统上。感兴趣的可以去网站下载安装包,尝试一下。当然,前提是你要有可用的GPU。
后面有时间再详细介绍一下这些年GPU在数值模式方面的应用情况。
参考链接:1. https://wrfg.net/
- End -