Performance of NVIDIA Earth-2 Correction Diffusion NIM

Use this documentation for details about the performance of the NVIDIA Earth-2 Correction Diffusion (CorrDiff) NIM.

Evaluation Process

Performance of the CorrDiff NIM is measured by the time for an inference request to complete for a single sample. Two tests are performed, each with a different number of diffusion steps. The first test uses 8 steps, and represents a low-quality prediction. The second test uses 18 steps, and represents a high-quality prediction. Both the NIM and client share the same computer to minimize network latency. The performance is measured on several supported GPUs and averaged across three different runs.

Performance Results

The performance results appear in the following table.

Important

The measurements that appear here can deviate between computers and runs. The performance values depend on the specific hardware configuration that the NIM and client are running on.

Diffusion Steps	L40s	RTX6000	A100	H100
8 Steps	45.27s	38.52s	25.10s	17.66s
18 Steps	95.46s	81.71s	51.43s	32.04s