Skip to content

Installation

subdomain can currently be installed from GitHub via

pip install git+https://github.com/HiDiHlabs/SubDomain.git

GPU support

For most users GPU support can be installed via

pip install 'subdomain[cuda12] @ git+https://github.com/HiDiHlabs/SubDomain.git' --extra-index-url=https://pypi.nvidia.com

However, this may not work in all circumstances (depending on the system setup and available CUDA installations). Therefore, we recommend to follow the respective install instructions

  • JAX for accelerated convolutions (including TPU)
  • RAPIDS (only cuML is required) for accelerated KMeans