Installation#

torchgeo-bench targets Python 3.12 or newer. We recommend installing inside a fresh virtual environment using uv, which is the project’s canonical workflow.

uv (recommended)#

Clone the repository and let uv create the environment and install all development dependencies:

$ git clone https://github.com/torchgeo/torchgeo-bench.git
$ cd torchgeo-bench
$ uv sync --extra dev

The torchgeo-bench console script is then available via uv run:

$ uv run torchgeo-bench --help

pip#

If you prefer pip (e.g. inside a conda environment), install the project in editable mode:

$ git clone https://github.com/torchgeo/torchgeo-bench.git
$ cd torchgeo-bench
$ pip install -e ".[dev]"

Optional extras#

A number of dependencies are gated behind PEP 621 extras so that lean benchmarking installations do not pull in heavyweight libraries by default. Combine extras with comma-separated lists:

$ uv sync --extra docs --extra viz
$ # or
$ pip install -e ".[docs,viz]"

Extra	Pulls in
`dev`	ruff, pytest, pre-commit, mdformat, pyproject-fmt
`docs`	sphinx, pydata-sphinx-theme, myst-parser, sphinx-copybutton
`id`	`torchid` for intrinsic-dimension metrics (Python ≥ 3.13)
`olmoearth`	`olmoearth-pretrain-minimal` for the OlmoEarth backbone
`sam3`	`transformers` for the SAM 3 encoder
`viz`	`matplotlib`, `pillow` for segmentation visualisations
`all`	every extra above

Datasets are not installed by these extras — see Datasets for how to download GeoBench V1 / V2 with the bundled torchgeo-bench download command.