ai models graphcast

ai-models-graphcast

ai-models-graphcast é um plugin ai-models para executar o GraphCast do Google Deepmind.

GraphCast: Aprendendo previsão meteorológica global de médio alcance, pré-impressão arXiv: 2212.12794, 2022. https://arxiv.org/abs/2212.12794

GraphCast foi criado por 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 e Peter Battaglia.

Os pesos do modelo são disponibilizados para uso sob os termos da Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0). Você pode obter uma cópia da Licença em: https://creativecommons.org/licenses/by-nc-sa/4.0/.

Instalação

Para instalar o pacote, execute:

 pip instalar ai-models-graphcast

Isso instalará o pacote e a maioria de suas dependências.

Em seguida, para instalar dependências de graphcast (e Jax na GPU):

Graphcast e Jax

Graphcast depende do Jax, que precisa de instruções de instalação especiais para seu hardware específico.

Consulte o guia de instalação para seguir as instruções corretas.

Preparamos dois requirements.txt que você pode usar. Uma versão CPU e uma GPU:

Para o uso preferencial de GPU:

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

Para o uso mais lento da CPU:

pip install -r requirements.txt