Docker Setup

This folder provides reference Docker images for LEANN.

Build (default)

Default image (no forced CPU index):

docker build -f docker/Dockerfile docker -t leann:latest

CPU-optimized image (forces PyTorch CPU index):

docker build \
  --build-arg PYTHON_VERSION=3.12 \
  --build-arg LEANN_VERSION=0.3.6 \
  -f docker/Dockerfile.cpu docker -t leann:cpu

Development image (source tree + build/test toolchain + uv sync):

docker build -f docker/Dockerfile.dev . -t leann:dev

docker run --rm leann:latest
docker run --rm leann:cpu
docker run --rm -it leann:dev

Expected output:

LEANN installed and importable.

Both images keep LEANN package semantics unchanged (pip install leann with both backends).
Dockerfile.cpu uses the PyTorch CPU index to avoid downloading CUDA wheels.
Dockerfile.dev is for local development, not production deployment.