Commit ·
1b8913a
1
Parent(s): 547aa71
Updated installation guidelines.
Browse files- run_AIFS_v2.0.ipynb +12 -6
run_AIFS_v2.0.ipynb
CHANGED
|
@@ -112,7 +112,9 @@
|
|
| 112 |
"source": [
|
| 113 |
"If you have access to GPUs on ECMWF's ATOS HPCF, you can run this notebook through [JupyterHub](https://jupyterhub.ecmwf.int/). \n",
|
| 114 |
"\n",
|
| 115 |
-
"Select the **ECMWF ATOS HPCF - GPU partition**"
|
|
|
|
|
|
|
| 116 |
]
|
| 117 |
},
|
| 118 |
{
|
|
@@ -121,11 +123,10 @@
|
|
| 121 |
"metadata": {},
|
| 122 |
"outputs": [],
|
| 123 |
"source": [
|
| 124 |
-
"#
|
| 125 |
-
"#
|
| 126 |
-
"#
|
| 127 |
-
"#
|
| 128 |
-
"# !{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\""
|
| 129 |
]
|
| 130 |
},
|
| 131 |
{
|
|
@@ -336,6 +337,11 @@
|
|
| 336 |
"print(\"Initial date is\", DATE)"
|
| 337 |
]
|
| 338 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 339 |
{
|
| 340 |
"attachments": {},
|
| 341 |
"cell_type": "markdown",
|
|
|
|
| 112 |
"source": [
|
| 113 |
"If you have access to GPUs on ECMWF's ATOS HPCF, you can run this notebook through [JupyterHub](https://jupyterhub.ecmwf.int/). \n",
|
| 114 |
"\n",
|
| 115 |
+
"Select the **ECMWF ATOS HPCF - GPU partition**\n",
|
| 116 |
+
"\n",
|
| 117 |
+
"**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."
|
| 118 |
]
|
| 119 |
},
|
| 120 |
{
|
|
|
|
| 123 |
"metadata": {},
|
| 124 |
"outputs": [],
|
| 125 |
"source": [
|
| 126 |
+
"# %pip install anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3\n",
|
| 127 |
+
"# %pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
|
| 128 |
+
"# %pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
|
| 129 |
+
"# %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\""
|
|
|
|
| 130 |
]
|
| 131 |
},
|
| 132 |
{
|
|
|
|
| 337 |
"print(\"Initial date is\", DATE)"
|
| 338 |
]
|
| 339 |
},
|
| 340 |
+
{
|
| 341 |
+
"cell_type": "markdown",
|
| 342 |
+
"metadata": {},
|
| 343 |
+
"source": []
|
| 344 |
+
},
|
| 345 |
{
|
| 346 |
"attachments": {},
|
| 347 |
"cell_type": "markdown",
|