environment.yaml

name: fenv

channels:
  - pyg
  - pytorch
  - nvidia

dependencies:
  - python=3.11
  - pip

  # ML
  - pytorch::pytorch-cuda
  - pytorch::torchvision
  - pytorch::torchaudio
  - pyg::pyg
  - pyg::pytorch-sparse
  - pyg::pytorch-cluster
  - conda-forge::lightning
  - conda-forge::torchmetrics
  - nvidia::cudatoolkit
  - wandb

  # Math
  - numpy
  - scipy

  #Analysis
  - matplotlib
  - networkx
  - pandas

  # Config
  - colorlog
  - hydra-core
  - pyyaml

  # Testing
  - pre-commit
  - pytest
  - pytest-xdist
  - typeguard

  # Other
  - tqdm

  - pip:
      - hydra-colorlog