From 4951c524b5c7eca3a124800956bfdf2f5ccd3d65 Mon Sep 17 00:00:00 2001 From: Hubert Siejkowski Date: Fri, 15 Sep 2023 11:58:02 +0200 Subject: [PATCH] Initial import --- epos-ai-tools.yml | 10 ++++++++++ epos-ai-train.yml | 22 ++++++++++++++++++++++ 2 files changed, 32 insertions(+) create mode 100644 epos-ai-tools.yml create mode 100644 epos-ai-train.yml diff --git a/epos-ai-tools.yml b/epos-ai-tools.yml new file mode 100644 index 0000000..8348a6f --- /dev/null +++ b/epos-ai-tools.yml @@ -0,0 +1,10 @@ +name: epos-ai-tools +channels: + - conda-forge + - defaults +dependencies: + - python=3.9.13 + - pip + - pip: + - --extra-index-url https://epos-apps.grid.cyfronet.pl/api/packages/epos-ai/pypi/simple + - gpd_tool diff --git a/epos-ai-train.yml b/epos-ai-train.yml new file mode 100644 index 0000000..0ecd808 --- /dev/null +++ b/epos-ai-train.yml @@ -0,0 +1,22 @@ +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