ai models graphcast

ai-modelos-graphcast

ai-models-graphcast es un complemento de ai-models para ejecutar GraphCast de Google Deepmind.

GraphCast: Aprendizaje hábil de pronósticos meteorológicos globales de mediano plazo, preimpresión de arXiv: 2212.12794, 2022. https://arxiv.org/abs/2212.12794

GraphCast fue creado por Remi Lam, Álvaro Sánchez-González, 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 y Peter Battaglia.

Los pesos de los modelos están disponibles para su uso según los términos de Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0). Puede obtener una copia de la Licencia en: https://creativecommons.org/licenses/by-nc-sa/4.0/.

Instalación

Para instalar el paquete, ejecute:

 instalación de pip ai-modelos-graphcast

Esto instalará el paquete y la mayoría de sus dependencias.

Luego, para instalar las dependencias de Graphcast (y Jax en GPU):

Graphcast y Jax

Graphcast depende de Jax, que necesita instrucciones de instalación especiales para su hardware específico.

Consulte la guía de instalación para seguir las instrucciones correctas.

Hemos preparado dos requirements.txt que puedes utilizar. Una versión de CPU y GPU:

Para el uso preferido de GPU:

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

Para el uso más lento de la CPU:

pip install -r requirements.txt