platform-demo-scripts/epos-ai-train.yml

23 lines
351 B
YAML
Raw Normal View History

2023-09-20 22:38:58 +02:00
name: epos-ai-train
2023-09-08 09:51:52 +02:00
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
- jupyterlab
- notebook
- pip
- pip:
- seisbench==0.4.1