AI / docs /DockerDeployment.md
PBThuong's picture
Deploy LightRAG AI Assistant with Git LFS for binary assets and fonts
979853c
|
Raw
History Blame Contribute Delete
8.82 kB

LightRAG Docker Deployment

A lightweight Knowledge Graph Retrieval-Augmented Generation system with multiple LLM backend support.

🚀 Preparation

Clone the repository:

# Linux/MacOS
git clone https://github.com/HKUDS/LightRAG.git
cd LightRAG
# Windows PowerShell
git clone https://github.com/HKUDS/LightRAG.git
cd LightRAG

Configure your environment:

# Linux/MacOS
cp .env.example .env
# Edit .env with your preferred configuration
# Windows PowerShell
Copy-Item .env.example .env
# Edit .env with your preferred configuration

LightRAG can be configured using environment variables in the .env file:

Server Configuration

  • HOST: Server host (default: 0.0.0.0)
  • PORT: Server port (default: 9621)

LLM Configuration

  • LLM_BINDING: LLM backend to use (lollms/ollama/openai)
  • LLM_BINDING_HOST: LLM server host URL
  • LLM_MODEL: Model name to use

Embedding Configuration

  • EMBEDDING_BINDING: Embedding backend (lollms/ollama/openai)
  • EMBEDDING_BINDING_HOST: Embedding server host URL
  • EMBEDDING_MODEL: Embedding model name

RAG Configuration

  • MAX_ASYNC: Maximum async operations
  • MAX_TOKENS: Maximum token size
  • EMBEDDING_DIM: Embedding dimensions

🐳 Docker Deployment

Docker instructions work the same on all platforms with Docker Desktop installed.

Build Optimization

The Dockerfile uses BuildKit cache mounts to significantly improve build performance:

  • Automatic cache management: BuildKit is automatically enabled via # syntax=docker/dockerfile:1 directive
  • Faster rebuilds: Only downloads changed dependencies when uv.lock or bun.lock files are modified
  • Efficient package caching: UV and Bun package downloads are cached across builds
  • No manual configuration needed: Works out of the box in Docker Compose and GitHub Actions

Start LightRAG server:

docker compose up -d

If you used the interactive setup, start the generated stack with:

docker compose -f docker-compose.final.yml up -d

The interactive setup keeps .env host-usable. Container-only hostnames such as postgres or host.docker.internal, along with staged SSL paths under /app/data/certs/, are injected into the generated docker-compose.final.yml for the lightrag service instead of being persisted back into .env. On reruns, unchanged wizard-managed service blocks in docker-compose.final.yml are preserved by default. To repair or fully regenerate those managed blocks from the bundled templates, rerun the matching setup target with make env-base-rewrite or make env-storage-rewrite.

If the generated stack includes local Milvus, compose resolves MINIO_ACCESS_KEY_ID and MINIO_SECRET_ACCESS_KEY at startup from the repo .env or exported shell environment. The generated compose file does not snapshot those values, and docker compose exits immediately if either variable is missing.

Before exposing the generated stack beyond localhost, run:

make env-security-check

That command audits the current .env for missing authentication, unsafe whitelist settings, weak JWT secrets, and other setup-level security risks without rewriting any files.

LightRAG Server uses the following paths for data storage:

data/
├── rag_storage/    # RAG data persistence
└── inputs/         # Input documents

Optional: local vLLM embedding and reranker

To run embedding and/or reranking locally with vLLM, run make env-base and answer yes when prompted to run the embedding model and rerank service locally via Docker. That configures the embedding service to use BAAI/bge-m3 on port 8001 with a local vLLM server, and can also add a vllm-rerank service on port 8000.

Alternatively, rerun make env-base later and enable only the rerank Docker prompt to add the vllm-rerank service automatically. vLLM provides a v1/rerank endpoint that works with the cohere binding.

Example docker-compose.override.yml for GPU hosts (embedding + reranker):

services:
  vllm-embed:
    image: vllm/vllm-openai:latest
    runtime: nvidia
    command: >
      --model BAAI/bge-m3
      --port 8001
      --dtype float16
    ports:
      - "8001:8001"
    volumes:
      - ./data/hf-cache:/root/.cache/huggingface
    ipc: host
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

  vllm-rerank:
    image: vllm/vllm-openai:latest
    runtime: nvidia
    command: >
      --model BAAI/bge-reranker-v2-m3
      --port 8000
      --dtype float16
    ports:
      - "8000:8000"
    volumes:
      - ./data/hf-cache:/root/.cache/huggingface
    ipc: host
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

For CPU-only hosts, use the official CPU image instead:

services:
  vllm-embed:
    image: vllm/vllm-openai-cpu:latest
    command: >
      --model BAAI/bge-m3
      --port 8001
      --dtype float32
    ports:
      - "8001:8001"
    volumes:
      - ./data/hf-cache:/root/.cache/huggingface

  vllm-rerank:
    image: vllm/vllm-openai-cpu:latest
    command: >
      --model BAAI/bge-reranker-v2-m3
      --port 8000
      --dtype float32
    ports:
      - "8000:8000"
    volumes:
      - ./data/hf-cache:/root/.cache/huggingface

Add the embedding and rerank config to .env:

EMBEDDING_BINDING=openai
EMBEDDING_MODEL=BAAI/bge-m3
EMBEDDING_DIM=1024
EMBEDDING_BINDING_HOST=http://localhost:8001/v1
EMBEDDING_BINDING_API_KEY=local-key
VLLM_EMBED_DEVICE=cpu

RERANK_BINDING=cohere
RERANK_MODEL=BAAI/bge-reranker-v2-m3
RERANK_BINDING_HOST=http://localhost:8000/rerank
RERANK_BINDING_API_KEY=local-key
VLLM_RERANK_DEVICE=cpu

If LightRAG runs in Docker while vLLM runs on the host, the generated compose file rewrites those endpoints to:

EMBEDDING_BINDING_HOST=http://host.docker.internal:8001/v1
RERANK_BINDING_HOST=http://host.docker.internal:8000/rerank

For GPU, set:

VLLM_EMBED_DEVICE=cuda
VLLM_RERANK_DEVICE=cuda

Ensure the NVIDIA Container Toolkit is installed and the host has CUDA drivers available. The setup wizard uses the CPU image by default for cpu device and the GPU image for cuda device. When rerunning make env-base, an existing VLLM_EMBED_DEVICE / VLLM_RERANK_DEVICE value is preserved instead of being overwritten by a fresh GPU auto-detection result. Those templates already pin the matching vLLM --dtype (float32 on CPU, float16 on CUDA), so no separate VLLM_*_DTYPE environment variables are needed.

SSL certificates

The setup wizard stages TLS certificate files under ./data/certs/ before generating the compose file. This keeps generated host mounts under the same ./data root used by the default Docker deployment.

PostgreSQL image

The interactive setup defaults PostgreSQL to gzdaniel/postgres-for-rag:16.6. That image bundles both Apache AGE and pgvector so the generated stack works with PGGraphStorage and PGVectorStorage without extra extension setup.

Updates

To update the Docker container:

docker compose pull
docker compose down
docker compose up

Offline deployment

Software packages requiring transformers, torch, or cuda will is not preinstalled in the dokcer images. Consequently, document extraction tools such as Docling, as well as local LLM models like Hugging Face and LMDeploy, can not be used in an off line enviroment. These high-compute-resource-demanding services should not be integrated into LightRAG. Docling will be decoupled and deployed as a standalone service.

📦 Build Docker Images

For local development and testing

# Build and run with Docker Compose (BuildKit automatically enabled)
docker compose up --build

# Or explicitly enable BuildKit if needed
DOCKER_BUILDKIT=1 docker compose up --build

Note: BuildKit is automatically enabled by the # syntax=docker/dockerfile:1 directive in the Dockerfile, ensuring optimal caching performance.

For production release

multi-architecture build and push:

# Use the provided build script
./docker-build-push.sh

The build script will:

  • Check Docker registry login status
  • Create/use buildx builder automatically
  • Build for both AMD64 and ARM64 architectures
  • Push to GitHub Container Registry (ghcr.io)
  • Verify the multi-architecture manifest

Prerequisites:

Before building multi-architecture images, ensure you have:

  • Docker 20.10+ with Buildx support
  • Sufficient disk space (20GB+ recommended for offline image)
  • Registry access credentials (if pushing images)