ИИ-прогноз погоды на своём железе: запускаем AIFS 2.0 от ECMWF

Европейский центр среднесрочных прогнозов погоды (ECMWF) выпустил открытую модель AIFS Single 2.0, которая умеет предсказывать погоду с высокой точностью. Но, как часто бывает с современными нейросетями, получить открытые веса — лишь половина дела. Для запуска требуется конкретный GPU и сложное окружение, например библиотека flash-attn, которая поддерживается далеко не на всех видеокартах. Многие энтузиасты, желающие опробовать модель на своём компьютере, сталкиваются с непреодолимыми препятствиями. Однако есть способ обойти эти ограничения и построить собственный ИИ-прогноз погоды даже без мощного графического ускорителя.

Один из самых простых вариантов — воспользоваться платформой Hugging Face Jobs. Она позволяет запустить модель в облаке, не заботясь о настройке окружения. Достаточно загрузить нужный скрипт, выбрать ресурсы и через несколько минут получить результат. Это удобно для тех, кто хочет быстро оценить возможности AIFS, не углубляясь в технические детали. Правда, облачные вычисления требуют оплаты, но для разовых экспериментов это вполне оправдано.

Тем, кто предпочитает всё держать под контролем, подойдёт локальный запуск. Ключевая проблема — зависимость от flash-attn. Этот модуль ускоряет работу модели, но его установка часто превращается в квест. К счастью, существует обходной путь: можно отключить flash-attn или заменить его альтернативной реализацией. После этого модель становится гораздо более дружелюбной к окружению. Как показали эксперименты, AIFS Single 2.0 способна работать даже на процессоре (CPU) и на чипах Apple с поддержкой MPS. Производительность, конечно, будет ниже, чем на топовом GPU, но сам факт, что современную модель погоды можно запустить на обычном ноутбуке, впечатляет.

Так что теперь каждый может попробовать себя в роли метеоролога и получить прогноз, построенный искусственным интеллектом, не покидая собственного дома. Достаточно следовать инструкциям, и собственный ИИ-прогноз погоды станет реальностью.