diff --git a/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst b/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst index 3cf8bca03..2e5d50d27 100644 --- a/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst +++ b/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst @@ -92,6 +92,10 @@ PyTorch inference performance testing docker pull rocm/pytorch:rocm6.2.3_ubuntu22.04_py3.10_pytorch_release_2.3.0_triton_llvm_reg_issue + .. note:: + + The Chai-1 benchmark uses a specifically selected Docker image using ROCm 6.2.3 and PyTorch 2.3.0 to address an accuracy issue. + .. container:: model-doc pyt_clip_inference 2. Use the following command to pull the `ROCm PyTorch Docker image `_ from Docker Hub.