NVIDIA Earth-2 FourCastNet NIM 性能#

使用本文档了解 NVIDIA Earth-2 FourCastNet NIM 性能的详细信息。

评估过程#

FourCastNet NIM 的 API 在生成预测数据时将其流式传输回,以便用户按顺序接收每个时间步。性能是根据完成一组三个不同预测所需的时间来衡量的。每个预测都使用 2020-01-01T00:00:00 的初始条件,并请求五个输出变量 ['t2m', 'z500', 't850', 'u10m', 'v10m']。每个预测的请求长度如下

  • 6 小时预测:6 小时 / 1 步

  • 5 天预测:120 小时 / 20 步

  • 10 天预测:240 小时 / 40 步

注意

NIM 支持的两种模型配置预计具有相似的性能值。

性能结果#

性能结果显示在下表中。延迟是以秒为单位的总预测时间,吞吐量是每秒平均预测步数。

重要提示

预计列出的测量值在机器和运行之间可能会有所偏差。性能值很大程度上取决于 NIM 和客户端运行的特定硬件配置。

H100#

6 小时

5 天

10 天

延迟

3.49秒

9.83秒

16.08秒

吞吐量

0.29 步/秒

2.03 步/秒

2.49 步/秒

A100#

6 小时

5 天

10 天

延迟

3.53秒

9.92秒

17.14秒

吞吐量

0.28 步/秒

2.01 步/秒

2.33 步/秒

L40S#

6 小时

5 天

10 天

延迟

3.86秒

14.54秒

25.82秒

吞吐量

0.26 步/秒

1.37 步/秒

1.55 步/秒

RTX6000#

6 小时

5 天

10 天

延迟

3.83秒

13.18秒

23.09秒

吞吐量

0.26 步/秒

1.52 步/秒

1.73 步/秒