Installation

Prerequisites

• Python 3.10 – 3.14

• pip or poetry

• A working PyTorch installation (CPU or CUDA). See the support matrix for tested versions.

Install from PyPI

pip install deeptab

Verify the installation:

import deeptab
print(deeptab.__version__)

Install from source

For development or to use unreleased features:

git clone https://github.com/OpenTabular/DeepTab
cd DeepTab
poetry install

See Contributing for the full development setup.

Optional: Mamba CUDA kernels

The default DeepTab Mamba implementation runs on any hardware. If you want the original optimised CUDA kernels (requires a compatible GPU and CUDA toolkit):

pip install mamba-ssm

GPU setup

If you need a specific PyTorch + CUDA combination, install PyTorch first following the official selector, then install DeepTab:

# Example: CUDA 11.8
pip install torch --index-url https://download.pytorch.org/whl/cu118
pip install deeptab