NVIDIA Earth2Studio: как собрать ансамблевый прогноз погоды за вечер

NVIDIA Earth2Studio — фреймворк для прогнозирования погоды с помощью ИИ. В новом туториале показано, как с нуля построить ансамблевый пайплайн: установить компоненты, сохранив CUDA-окружение Colab, загрузить предобученную модель FCN и получить начальные условия из глобальной системы GFS.

Автор добавил собственную диагностику: преобразование 10-метровых ветров в коэффициент мощности турбины (capacity factor), а также систему возмущений с разным уровнем шума для разных переменных. Ансамбль из 8 членов выполняется через низкоуровневые API Earth2Studio — итераторы, батчинг, координаты, Zarr.

Результаты записываются в координатно-ориентированное хранилище, затем верифицируются по анализу GFS: RMSE, CRPS, разброс ансамбля и отношение spread-skill. Визуализация включает пространственные карты, спагетти-диаграммы геопотенциала, веерные графики в точке (Нью-Дели) и кривые навыка по времени упреждения.

Полный код доступен в статье — подходит для воспроизведения в Google Colab с GPU T4. Это готовый шаблон для собственных экспериментов с ансамблевым прогнозом погоды на нейросетях.