env.yaml

name: nlam
channels:
  - pytorch
  - nvidia
  - pyg
  - conda-forge
channel_priority: strict  
dependencies:
  - python == 3.10.13
  - pytorch==2.2.2
  - pyg==2.5.2  
  - torchvision==0.17.2
  - pytorch-cuda==12.1
  - numpy>=1.24.2
  - wandb>=0.13.10
  - matplotlib>=3.7.0
  - scipy>=1.10.0
  - pytorch-lightning>=2.1.2
  - lightning==2.2.4
  - shapely>=2.0.1
  - networkx>=3.0
  - Cartopy>=0.22.0
  - pyproj>=3.4.1
  - plotly>=5.15.0
  - tornado>=6.3.3
  - cython>=3
  - cfgrib==0.9.11.0
  - dataclasses-json==0.6.4
  - xarray==2024.5.0
  - argparse-dataclass==2.0.0
  - tensorboard==2.13.0
  - typer==0.9.0
  - netCDF4==1.6.5
  - einops==0.7.0
  - torchinfo==1.8.0
  - tabulate==0.9.0
  - pytest==8.1.1
  - onnx==1.16.0
  - onnxruntime==1.17.3
  - monai==1.3.1
  - pip
  - pip:
     - tensorboard-plugin-profile
     - torch-tb-profiler==0.4.1
     - tueplots>=0.0.8   
     - onnxscript==0.1.0.dev20240421
     - transformers==4.44.2
     - gif==23.3.0
     - scikit-image==0.24.0