diff --git a/README.md b/README.md index 10fce5b..395341b 100644 --- a/README.md +++ b/README.md @@ -19,13 +19,21 @@ This code is based on the [pick-benchmark](https://github.com/seisbench/pick-ben Please download and install [Mambaforge](https://github.com/conda-forge/miniforge#mambaforge) following the [official guide](https://github.com/conda-forge/miniforge#install). -After successful installation and within the Mambaforge environment please run following commands: +After successful installation and within the Mambaforge environment please clone this repository: ``` git clone ssh://git@git.plgrid.pl:7999/eai/platform-demo-scripts.git +``` +and please run for Linux or Windows platforms: +``` cd platform-demo-scripts -mambaforge env create -f environment.yml +mambaforge env create -f epos-ai-train.yml +``` +or for OSX: +``` +cd platform-demo-scripts +mambaforge env create -f epos-ai-train-osx.yml ``` This will create a conda environment named `platform-demo-scripts` with all required packages installed. diff --git a/epos-ai-train-osx.yml b/epos-ai-train-osx.yml new file mode 100644 index 0000000..1502139 --- /dev/null +++ b/epos-ai-train-osx.yml @@ -0,0 +1,21 @@ +name: epos-ai-train +channels: + - pytorch + - nvidia +dependencies: + - python=3.10 + - pytorch=2.0.1 + - 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 diff --git a/environment.yml b/epos-ai-train.yml similarity index 92% rename from environment.yml rename to epos-ai-train.yml index 6d2b921..0ecd808 100644 --- a/environment.yml +++ b/epos-ai-train.yml @@ -1,4 +1,4 @@ -name: platform-demo-scripts +name: epos-ai-train channels: - pytorch - nvidia