ai models graphcast

AI-модели-графкаст

ai-models-graphcast — это плагин ai-models для запуска GraphCast от Google Deepmind.

GraphCast: Обучение умелому среднесрочному глобальному прогнозу погоды, препринт arXiv: 2212.12794, 2022. https://arxiv.org/abs/2212.12794.

GraphCast создали Реми Лам, Альваро Санчес-Гонсалес, Мэтью Уилсон, Питер Вирнсбергер, Мейре Фортунато, Ферран Алет, Суман Равури, Тимо Эвальдс, Зак Итон-Розен, Вейхуа Ху, Александр Мероз, Стефан Хойер, Джордж Холланд, Ориол Виньялс, Жаклин Стотт, Александр Притцель, Шакир Мохамед и Питер Батталья.

Веса моделей доступны для использования в соответствии с условиями Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0). Вы можете получить копию Лицензии по адресу: https://creativecommons.org/licenses/by-nc-sa/4.0/.

Установка

Чтобы установить пакет, запустите:

 pip install ai-models-graphcast

При этом будет установлен пакет и большинство его зависимостей.

Затем, чтобы установить зависимости графкаста (и Jax на графическом процессоре):

Графкаст и Джакс

Graphcast зависит от Jax, для которого требуются специальные инструкции по установке для вашего конкретного оборудования.

Пожалуйста, ознакомьтесь с руководством по установке, чтобы следовать правильным инструкциям.

Мы подготовили два requirements.txt которые вы можете использовать. Версия процессора и графического процессора:

Для предпочтительного использования графического процессора:

pip install -r requirements-gpu.txt -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

Для более медленного использования процессора:

pip install -r requirements.txt