Installation

The package requires Python 3.12. The package can be installed using the following commands. This will install the package with the CPU dependencies, i.e., the cpu version of PyTorch and scikit-learn for clustering.

Create a virtual environment, e.g. using conda.

conda create -n deepgeodemo python=3.12
conda activate deepgeodemo

Install the deepgeodemo package via pip.You might want to pre-install PyTorch using the CPU-only wheel to avoid unnecessary dependencies being downloaded.

pip install torch --index-url https://download.pytorch.org/whl/cpu
pip install deepgeodemo

Alternatively, the package can be installed with the GPU dependencies if available and RAPIDS for the clustering.

pip install --extra-index-url=https://pypi.nvidia.com deepgeodemo[gpu]