// create conda environment within pycharm, python 3.10.8

conda install mamba -n base -c conda-forge
mamba install nvidia/label/cuda-12.4.0::cuda-toolkit

// pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu124
mamba install pytorch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 pytorch-cuda=12.4 -c pytorch -c nvidia
// mamba install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.4 -c pytorch -c nvidia

// mamba install pyg -c pyg
pip install torch_geometric
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.4.0+cu124.html

pip install --user torchmetrics hydra-core einops matplotlib scikit-image opencv-python pandas colorlog
pip install wandb

pip install -e .