name: epos-ai-train
channels:
  - pytorch
  - nvidia
dependencies:
  - python=3.10
  - pytorch=2.0.1
  - pytorch-cuda
  - PyYAML=6.0
  - python-dotenv=1.0.0
  - pandas=2.0.3
  - obspy=1.4.0
  - wandb=0.15.4
  - torchmetrics=0.11.4
  - pytorch-lightning
  - scikit-learn
  - openpyxl
  - pydantic
  - jupyterlab
  - notebook
  - pip
  - pip:
    - seisbench==0.4.1