Installation

We recommend CUDA 12.1 with cu* packages version ≥24.2 and <26.0. Ensure your CUDA driver version matches or exceeds your toolkit version (≥12.1 for CUDA 12.1). Adjust package versions in the environment files below if your system requires a different package versions.

Clone the repository

git clone https://github.com/dpeerlab/segger.git segger && cd segger

Using conda

conda env create -n segger -f environment_cuda121.yml

Adjust environment_cuda121.yml for other CUDA versions (e.g., environment_cuda118.yml for CUDA 11.8).

Using pixi

pixi install -e cuda121

Adjust the environment name in pixi.toml as needed for other CUDA versions.

pip

Install GPU-accelerated PyTorch and RAPIDS compatible with your CUDA version before installing segger. All CUDA-enabled packages must be compiled for the same CUDA version.

• PyTorch & torchvision: Installation guide
• torch_scatter: Installation guide
• RAPIDS (cuDF, cuML, cuGraph): Installation guide
• CuPy: Installation guide
• cuSpatial: Installation guide

For example, on Linux with CUDA 12.1 and PyTorch 2.5.0:

# Install PyTorch and torchvision for CUDA 12.1
pip install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu121

# Install torch_scatter for CUDA 12.1
pip install torch_scatter -f https://data.pyg.org/whl/torch-2.5.0+cu121.html

# Install RAPIDS packages for CUDA 12.x
pip install --extra-index-url=https://pypi.nvidia.com cuspatial-cu12 cudf-cu12 cuml-cu12 cugraph-cu12

# Install CuPy for CUDA 12.x
pip install cupy-cuda12x

December 2025: To stay up-to-date with new developments, we recommend installing the latest version directly from GitHub:

# Clone segger repo and install locally
pip install -e .

Usage

You can run segger from the command line with:

segger segment -i /path/to/your/ist/data/ -o /path/to/save/outputs/

To see all available parameter options:

segger segment --help