mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-23 10:33:51 +00:00
22 lines
482 B
Docker
22 lines
482 B
Docker
FROM python:3.11-slim
|
|
|
|
WORKDIR /app
|
|
|
|
# Install required packages
|
|
RUN pip install --no-cache-dir \
|
|
aiohttp \
|
|
asyncio \
|
|
statistics
|
|
|
|
# Copy benchmark script
|
|
COPY vllm_llama3_benchmark.py /app/
|
|
|
|
# Create results directory
|
|
RUN mkdir -p /app/results
|
|
|
|
# Make script executable
|
|
RUN chmod +x /app/vllm_llama3_benchmark.py
|
|
|
|
# Default command
|
|
CMD ["python", "/app/vllm_llama3_benchmark.py", "--url", "http://vllm-llama3-8b:8001", "--output", "/app/results/benchmark_results.json"]
|