// create conda environment within pycharm, python 3.12

conda install mamba -n base -c conda-forge

//download the visual studio build tools
https://visualstudio.microsoft.com/pl/visual-cpp-build-tools/

mamba install nvidia/label/cuda-12.8.1::cuda-toolkit


## THIS SHOULD WORK BUT I DID NOT TESTED IT THOROUGHLY ##
pip install srforge[cuda128] -i https://pypi.org/simple --extra-index-url https://download.pytorch.org/whl/cu128 -f https://data.pyg.org/whl/torch-2.8.0+cu128.html
## OR THE ONE BELOW IF YOU WANT TO HAVE A MODIFIABLE INSTALLATION (YOU NEED THE SR-FORGE REPO CLONED TO THE DIRECTORY THAT IS PASSED AS THE SECOND ARGUMENT)
pip install -e ../sr-forge[cuda128] -i https://pypi.org/simple --extra-index-url https://download.pytorch.org/whl/cu128 -f https://data.pyg.org/whl/torch-2.7.0+cu128.html

## If the above command fails, try installing the dependencies manually ##
pip install torch_geometric
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.8.0+cu128.html

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

pip install -e .

