# clone project
git clone https://github.com/carbonscott/maxie
cd maxie

# [OPTION 1] create python virtual environment
python3 -m venv ./myenv
. ./myenv/bin/activate
export PIP_CACHE_DIR=/tmp/$USER

# [OPTION 2] create conda environment
conda create -n myenv python=3.11
conda activate myenv

# install pytorch according to instructions
# https://pytorch.org/get-started/

# install mpi4py and h5py
MPICC="cc -shared" pip install --no-cache-dir --no-binary=mpi4py mpi4py
HDF5_MPI="ON" CC=cc HDF5_DIR=${HDF5_ROOT} pip install --no-cache-dir --no-binary=h5py h5py

# install requirements
pip install -r requirements.txt

# install maxie (-e for editable install)
pip install -e .

# train on CPU
python src/train.py trainer=cpu

# train on GPU
python src/train.py trainer=gpu

python src/train.py experiment=experiment_name.yaml

python src/train.py trainer.max_epochs=20 data.batch_size=64