Sign inSign up

holtmann/llm-benchmark

By holtmann

Updated 9 months ago

Benchmark suite for LLM inference on NVIDIA consumer GPUs (RTX 5060 Ti, 5070 Ti, 5090)

Image
Machine learning & AI
0

2.3K

holtmann/llm-benchmark repository overview

LLM Benchmark

Pre-configured environment for benchmarking LLM inference on NVIDIA GPUs with vLLM and AIPerf.

Quick Start

docker pull holtmann/llm-benchmark:latest

# Run interactively
docker run --gpus all -it holtmann/llm-benchmark:latest

# Start vLLM server
docker run --gpus all -p 8000:8000 holtmann/llm-benchmark:latest \
  python -m vllm.entrypoints.openai.api_server \
  --model nvidia/Qwen3-8B-NVFP4 \
  --max-model-len 8192

What's Included

ComponentVersionPurpose
vLLM0.12.0High-throughput LLM serving with NVFP4/MXFP4 support
AIPerf0.3.0LLM inference benchmarking
lm-eval0.4.9.1Language model evaluation harness
DCGM4.4.2GPU telemetry and power monitoring
CUDA12.9GPU compute (devel image for FP8 KV cache JIT)

Supported GPUs

Optimized for NVIDIA Blackwell architecture (RTX 50 series) with full support for:

  • NVFP4 quantization (weights + activations)
  • MXFP4 quantization
  • FP8 KV cache

Also works on Ampere, Ada Lovelace, and Hopper GPUs.

Environment Variables

VariableDescription
HF_TOKENHuggingFace token for gated models (Gemma, Llama, etc.)
CUDA_VISIBLE_DEVICESGPU selection

Exposed Ports

PortService
8000vLLM OpenAI-compatible API
9400DCGM Exporter (Prometheus metrics)

Example: Run Benchmark

# Start container with GPU access
docker run --gpus all -it \
  -e HF_TOKEN=your_token \
  -v $(pwd)/results:/workspace/results \
  holtmann/llm-benchmark:latest

# Inside container: start vLLM
python -m vllm.entrypoints.openai.api_server \
  --model nvidia/Qwen3-8B-NVFP4 \
  --max-model-len 8192 \
  --gpu-memory-utilization 0.9 &

# Run AIPerf benchmark
aiperf \
  --endpoint-type chat \
  --base-url http://localhost:8000 \
  --model nvidia/Qwen3-8B-NVFP4 \
  --streaming \
  --concurrency 8 \
  --num-requests 100 \
  --synthetic-input-tokens-mean 8192 \
  --output-tokens-mean 512

Example: Quality Evaluation

# Run lm-eval benchmarks
lm_eval --model vllm \
  --model_args pretrained=nvidia/Qwen3-8B-NVFP4 \
  --tasks mmlu,gsm8k,hellaswag \
  --batch_size auto

GPU Monitoring

# Start DCGM monitoring (inside container)
dcgmi dmon -e 155,156,203,204 -d 1000

# Fields: Power (W), Temperature (C), GPU Util (%), Memory Util (%)

Tags

TagDescription
latestMost recent stable build

Source Code

Research

This image was used for benchmarking RTX 5060 Ti, 5070 Ti, and 5090 GPUs across RAG, API, and Agentic workloads. Full results and methodology available in the GitHub repository.

License

MIT License

Tag summary

Content type

Image

Digest

sha256:7c4ed78b1

Size

11 GB

Last updated

9 months ago

docker pull holtmann/llm-benchmark