environment.yml

name: waterfutures
channels:
  - pyg
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - cudatoolkit=11.8.0=h6a678d5_0
  - numpy=1.26.4=py39h5f9d8c6_0
  - pandas=2.2.2=py39h6a678d5_0
  - pygad=3.3.1=pyhd8ed1ab_0
  - python=3.9.18=h955ad1f_0
  - pytorch=2.2.1=py3.9_cuda12.1_cudnn8.9.2_0
  - pytorch-scatter=2.1.2=py39_torch_2.2.0_cu121
  - pyyaml=6.0.1=py39h5eee18b_0
  - tabulate=0.9.0=py39h06a4308_0
  - torchaudio=2.2.1=py39_cu121
  - torchmetrics=1.3.1=pyhd8ed1ab_0
  - torchtriton=2.2.0=py39
  - torchvision=0.17.1=py39_cu121
  - tqdm=4.66.2=pyhd8ed1ab_0
  - xlsxwriter=3.1.1=py39h06a4308_0
  - yaml=0.2.5=h7b6447c_0
  - pip:
      - matplotlib==3.9.1
      - networkx==3.2.1
      - optuna==3.6.1
      - python-calamine==0.2.0
      - ray==2.34.0
      - scikit-learn==1.4.1.post1
      - scipy==1.13.1
      - seaborn==0.13.2
      - torch-geometric==2.5.0
      - wandb==0.16.4
      - watchfiles==0.22.0
      - wntr==1.2.0