ai models graphcast

ai-models-graphcast

ai-models-graphcast est un plugin ai-models pour exécuter GraphCast de Google Deepmind.

GraphCast : Apprendre des prévisions météorologiques mondiales à moyen terme, préimpression arXiv : 2212.12794, 2022. https://arxiv.org/abs/2212.12794

GraphCast a été créé par Remi Lam, Alvaro Sanchez-Gonzalez, Matthew Willson, Peter Wirnsberger, Meire Fortunato, Ferran Alet, Suman Ravuri, Timo Ewalds, Zach Eaton-Rosen, Weihua Hu, Alexander Merose, Stephan Hoyer, George Holland, Oriol Vinyals, Jacklynn Stott, Alexander Pritzel, Shakir Mohamed et Peter Battaglia.

Les pondérations du modèle sont mises à disposition pour utilisation selon les termes de Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0). Vous pouvez obtenir une copie de la licence à l'adresse : https://creativecommons.org/licenses/by-nc-sa/4.0/.

Installation

Pour installer le package, exécutez :

 pip installer ai-models-graphcast

Cela installera le package et la plupart de ses dépendances.

Ensuite pour installer les dépendances graphcast (et Jax sur GPU) :

Graphcast et Jax

Graphcast dépend de Jax, qui nécessite des instructions d'installation spéciales pour votre matériel spécifique.

Veuillez consulter le guide d'installation pour suivre les instructions correctes.

Nous avons préparé deux requirements.txt que vous pouvez utiliser. Une version CPU et une version GPU :

Pour l’utilisation préférée du GPU :

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

Pour une utilisation plus lente du processeur :

pip install -r requirements.txt