23 lines
351 B
YAML
23 lines
351 B
YAML
|
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
|
||
|
|
- jupyterlab
|
||
|
|
- notebook
|
||
|
|
- pip
|
||
|
|
- pip:
|
||
|
|
- seisbench==0.4.1
|