Sign inSign up

holtmann/llm-benchmark

By holtmann

•Updated 10 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

10 months ago

docker pull holtmann/llm-benchmark