ai-models-containers

(In development) A collection of Dockerfiles to build containers for running ECMWF ai-models

Notes

• Requires NVIDIA Container Toolkit to build for GPU support
• Current containers built for CUDA Version 12.3. You may need to modify source image to match CUDA version to your GPU supported version (i.e 12.3, 12.4, etc..)
• Graphcast untested because of lack of access to GPU with proper memory
• GPU Memory Requirements:
  • FourCastNet GPU ~ 12 GB
  • PanguWeather ~ 32 GB
  • Graphcast 0.25 Degree ~ 40 GB
  • Graphcast 1.0 Degree ~ 32 GB

Build container for FourCastNet and PanguWeather

docker build --rm --pull -t ai-models - < ai-models.Dockerfile

Build container for Graphcast

docker build --rm --pull -t ai-models-graphcast - < graphcast.Dockerfile

Example running FourCastNet using ECMWF CDS for initial conditions:

(modify paths to assets, outputs, and cdsapirc file to your environment)

docker run --rm -it -e AI_MODELS_ASSETS=/assets/ -v /home/user/ai-models/assets/:/assets -v /home/user/ai-models/output/:/output -v /home/user/.cdsapirc:/root/.cdsapirc --gpus all ai-models:latest ai-models --download-assets --input cds --assets-sub-directory --date 20240713 --path /output/output-{step}.grb fourcastnetv2-small

Example running FourCastNet using GFS for initial conditions:

(modify paths to assets, outputs, and cdsapirc file to your environment)

docker run --rm -it -e AI_MODELS_ASSETS=/assets/ -v /home/user/ai-models/assets/:/assets -v /home/user/ai-models/output/:/output -v /home/user/.cdsapirc:/root/.cdsapirc --gpus all ai-models:latest ai-models-gfs --download-assets --input gfs --assets-sub-directory --date 20240713 --path /output/output-{step}.grb fourcastnetv2-small

Example running Graphcast at 1 degree resolution using GFS for initial conditions:

(modify paths to assets, outputs, and cdsapirc file to your environment)

docker run --rm -it -e AI_MODELS_ASSETS=/assets/ -v /home/user/ai-models/assets/:/assets -v /home/user/ai-models/output/:/output -v /home/user/.cdsapirc:/root/.cdsapirc --gpus all ai-models-graphcast:latest ai-models-gfs --download-assets --input gfs --assets-sub-directory --date 20240713 --path /output/output-{step}.grb --onedeg graphcast

To Do

• Add GFS input

Disclaimer

This repository is a scientific product and is not official communication of the National Oceanic and Atmospheric Administration, or the United States Department of Commerce. All NOAA GitHub project code is provided on an “as is” basis and the user assumes responsibility for its use. Any claims against the Department of Commerce or Department of Commerce bureaus stemming from the use of this GitHub project will be governed by all applicable Federal law. Any reference to specific commercia products, processes, or services by service mark, trademark, manufacturer, or otherwise, does not constitute or imply their endorsement, recommendation or favoring by the Department of Commerce. The Department of Commerce seal and logo, or the seal and logo of a DOC bureau, shall not be used in any manner to imply endorsement of any commercial product or activity by DOC or the United States Government.