Instructions to use ecmwf/aifs-ens-2.0 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- AnemoI
How to use ecmwf/aifs-ens-2.0 with AnemoI:
from anemoi.inference.runners.default import DefaultRunner from anemoi.inference.config.run import RunConfiguration # Create Configuration config = RunConfiguration(checkpoint = {"huggingface":"ecmwf/aifs-ens-2.0"}) # Load Runner runner = DefaultRunner(config) - Notebooks
- Google Colab
- Kaggle
Commit ·
b5eef80
1
Parent(s): 49e7db4
Updated installation guidelines.
Browse files- run_AIFS_ENS_v2.0.ipynb +7 -6
run_AIFS_ENS_v2.0.ipynb
CHANGED
|
@@ -109,7 +109,9 @@
|
|
| 109 |
"source": [
|
| 110 |
"If you have access to GPUs on ECMWF's ATOS HPCF, you can run this notebook through [JupyterHub](https://jupyterhub.ecmwf.int/). \n",
|
| 111 |
"\n",
|
| 112 |
-
"Select the **ECMWF ATOS HPCF - GPU partition**"
|
|
|
|
|
|
|
| 113 |
]
|
| 114 |
},
|
| 115 |
{
|
|
@@ -119,11 +121,10 @@
|
|
| 119 |
"metadata": {},
|
| 120 |
"outputs": [],
|
| 121 |
"source": [
|
| 122 |
-
"#
|
| 123 |
-
"#
|
| 124 |
-
"#
|
| 125 |
-
"#
|
| 126 |
-
"# !{sys.executable} -m pip install -q --force-reinstall --no-deps \"flash-attn @ https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl\""
|
| 127 |
]
|
| 128 |
},
|
| 129 |
{
|
|
|
|
| 109 |
"source": [
|
| 110 |
"If you have access to GPUs on ECMWF's ATOS HPCF, you can run this notebook through [JupyterHub](https://jupyterhub.ecmwf.int/). \n",
|
| 111 |
"\n",
|
| 112 |
+
"Select the **ECMWF ATOS HPCF - GPU partition**\n",
|
| 113 |
+
"\n",
|
| 114 |
+
"**Note:** The quick-start method below installs packages in your personal Python package directory for the active Jupyter kernel. This is convenient for running the notebook, but it may affect other notebooks that use the same Python environment. We recommend creating a dedicated Python environment and registering it as a Jupyter kernel."
|
| 115 |
]
|
| 116 |
},
|
| 117 |
{
|
|
|
|
| 121 |
"metadata": {},
|
| 122 |
"outputs": [],
|
| 123 |
"source": [
|
| 124 |
+
"# %pip install anemoi-inference[huggingface]==0.8.3 anemoi-models==0.11.2\n",
|
| 125 |
+
"# %pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
|
| 126 |
+
"# %pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
|
| 127 |
+
"# %ip install -q --force-reinstall --no-deps \"flash-attn @ https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl\""
|
|
|
|
| 128 |
]
|
| 129 |
},
|
| 130 |
{
|