Skip to main content

Run graphcast ai weather models with capabilities for GFS and GDAS initial conditions and NetCDF output

Project description

ai-models-graphcast-gfs

ai-models-graphcast-gfs is an extension of the ai-models-graphcast package, which itself is a plugin to run Google Deepmind's GraphCast with ai-models.

GraphCast: Learning skillful medium-range global weather forecasting, arXiv preprint: 2212.12794, 2022. https://arxiv.org/abs/2212.12794

GraphCast was created by 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 and Peter Battaglia.

The model weights are made available for use under the terms of the Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0). You may obtain a copy of the License at: https://creativecommons.org/licenses/by-nc-sa/4.0/.

Installation

To install the package, run:

pip install ai-models-graphcast-gfs

This will install the package and most of its dependencies.

Then to install graphcast dependencies (and Jax on GPU):

Graphcast and Jax

Graphcast depends on Jax, which needs special installation instructions for your specific hardware.

Please see the installation guide to follow the correct instructions.

We have prepared two requirements.txt you can use. A CPU and a GPU version:

For the preferred GPU usage:

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

For the slower CPU usage:

pip install -r requirements.txt

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ai_models_graphcast_gfs-0.0.9.tar.gz (15.8 kB view details)

Uploaded Source

Built Distribution

ai_models_graphcast_gfs-0.0.9-py3-none-any.whl (17.8 kB view details)

Uploaded Python 3

File details

Details for the file ai_models_graphcast_gfs-0.0.9.tar.gz.

File metadata

File hashes

Hashes for ai_models_graphcast_gfs-0.0.9.tar.gz
Algorithm Hash digest
SHA256 05c21c058e20db3a01b07143aa2f56c8e16dd1bec7c5c87dfc584400daf4189f
MD5 9ae76692c1b32dd6d326aefb317e6b78
BLAKE2b-256 ac64e15cf46989b30e755c3382360351497e388b760a394d6ed9ff5a5c8b1f11

See more details on using hashes here.

File details

Details for the file ai_models_graphcast_gfs-0.0.9-py3-none-any.whl.

File metadata

File hashes

Hashes for ai_models_graphcast_gfs-0.0.9-py3-none-any.whl
Algorithm Hash digest
SHA256 c0f6b16230cd98ad4a36f63e31564f0704446d71c5486348572224d64e66bc6e
MD5 b4e3cae97db82b4fb982072d8f853c73
BLAKE2b-256 d1a87e80020e096582a9e9301938fe985190574a485126e2a660e5fc01078021

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page