Pull the AI Runner Docker Image
Pull Pipeline-Specific Images (optional)
Download AI Models
Execute the Benchmark Test
<GPU_IDs>
: Specify which GPU(s) to use. For example, '"device=0"'
for GPU 0, '"device=0,1"'
for GPU 0 and GPU 1, or '"device=all"'
for all GPUs.<PIPELINE>
: The pipeline to benchmark (e.g., text-to-image
).<MODEL_ID>
: The model ID to use for benchmarking (e.g., stabilityai/sd-turbo
).<RUNS>
: The number of benchmark runs to perform.<NUM_INFERENCE_STEPS>
: The number of inference steps to perform (optional).Analyze the Benchmark Results