environment.yml

name: ecmwf-ml-course
channels:
    - pytorch
    - nvidia
    - pyg
    - plotly
    - defaults
    - conda-forge
dependencies:
    - python>=3.10
    - pip>=24.0
    - jupyterlab>=4
    - jupyterlab-git
    - jupyter-resource-usage
    - ipywidgets>=7.6.0
    - pytorch>=2.1.0
    - pytorch-cuda=11.8
    - scikit-learn>=1.4.0
    - numpy>=1.26.0
    - matplotlib>=3.8.0
    - seaborn>=0.13.0
    - plotly>=5.19.0
    - tqdm>=4.66.0
    - xarray>=2024.2.0
    - h3-py>=3.7.7
    - networkx>=3.2.0
    - keras>=3.0.0
    - torchvision>=0.9.0
    - torchaudio>=2.2.0
    - torchinfo>=1.8.0
    - py-xgboost>=2.0.0
    - pyg>=2.5.0
    - pytorch-scatter>=2.1.0
    - pytorch-sparse>=0.6.0
    - pytorch-cluster>=1.6.0
    - pandas>=1.5.0
    - shap>=0.44.0
    - pip:
          - climetlab>=0.21.0
          - climetlab-mltc-surface-observation-postprocessing>=0.3.1
          - tensorflow[and-cuda]>=2.15.0
          - quantus>=0.5.0
          - tf-explain