# ai-models-graphcast
`ai-models-graphcast` is an [ai-models](https://github.com/ecmwf-lab/ai-models) plugin to run Google Deepmind's [GraphCast](https://github.com/deepmind/graphcast).
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:
```bash
pip install ai-models-graphcast
```
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](https://github.com/google/jax#installation) 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
```
Raw data
{
"_id": null,
"home_page": "https://github.com/ecmwf-lab/ai-models-graphcast",
"name": "ai-models-graphcast",
"maintainer": "",
"docs_url": null,
"requires_python": "",
"maintainer_email": "",
"keywords": "tool",
"author": "European Centre for Medium-Range Weather Forecasts (ECMWF)",
"author_email": "software.support@ecmwf.int",
"download_url": "https://files.pythonhosted.org/packages/5d/d5/11bfd8b43c8c0cd44af1a4e9ad66405f5d6a234e34cf43915cfc2e88b52f/ai-models-graphcast-0.0.7.tar.gz",
"platform": null,
"description": "# ai-models-graphcast\n\n`ai-models-graphcast` is an [ai-models](https://github.com/ecmwf-lab/ai-models) plugin to run Google Deepmind's [GraphCast](https://github.com/deepmind/graphcast).\n\nGraphCast: Learning skillful medium-range global weather forecasting, arXiv preprint: 2212.12794, 2022. https://arxiv.org/abs/2212.12794\n\nGraphCast 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.\n\nThe 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/.\n\n## Installation\n\nTo install the package, run:\n\n```bash\npip install ai-models-graphcast\n```\n\nThis will install the package and most of its dependencies.\n\nThen to install graphcast dependencies (and Jax on GPU):\n\n\n\n### Graphcast and Jax\n\nGraphcast depends on Jax, which needs special installation instructions for your specific hardware.\n\nPlease see the [installation guide](https://github.com/google/jax#installation) to follow the correct instructions.\n\nWe have prepared two `requirements.txt` you can use. A CPU and a GPU version:\n\nFor the preferred GPU usage:\n```\npip install -r requirements-gpu.txt -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html\n```\n\nFor the slower CPU usage:\n```\npip install -r requirements.txt\n```\n",
"bugtrack_url": null,
"license": "Apache License Version 2.0",
"summary": "Run graphcast ai weather models",
"version": "0.0.7",
"project_urls": {
"Homepage": "https://github.com/ecmwf-lab/ai-models-graphcast"
},
"split_keywords": [
"tool"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "5dd511bfd8b43c8c0cd44af1a4e9ad66405f5d6a234e34cf43915cfc2e88b52f",
"md5": "d62b765cbc15550e4f38d20a3af102ed",
"sha256": "a22b5168ab8c92c9b5b0975c23db0079d3a4bfcfd1fbcd1e1b8b1be80d1da60d"
},
"downloads": -1,
"filename": "ai-models-graphcast-0.0.7.tar.gz",
"has_sig": false,
"md5_digest": "d62b765cbc15550e4f38d20a3af102ed",
"packagetype": "sdist",
"python_version": "source",
"requires_python": null,
"size": 12752,
"upload_time": "2024-02-13T07:47:13",
"upload_time_iso_8601": "2024-02-13T07:47:13.771622Z",
"url": "https://files.pythonhosted.org/packages/5d/d5/11bfd8b43c8c0cd44af1a4e9ad66405f5d6a234e34cf43915cfc2e88b52f/ai-models-graphcast-0.0.7.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-02-13 07:47:13",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "ecmwf-lab",
"github_project": "ai-models-graphcast",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"requirements": [
{
"name": null,
"specs": []
}
],
"tox": true,
"lcname": "ai-models-graphcast"
}
JSON
