github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-02-03 10:55:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add note for chai-1 benchmark Docker in pytorch-inference-benchmark.rst (#4684)

Browse Source
This commit is contained in:
Peter Park
2025-04-24 15:48:53 -04:00
committed by GitHub
parent 9293723381
commit 1f41ce26be
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst
View File

@@ -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 <https://hub.docker.com/layers/rocm/pytorch/rocm6.2.3_ubuntu22.04_py3.10_pytorch_release_2.3.0_triton_llvm_reg_issue/images/sha256-b736a4239ab38a9d0e448af6d4adca83b117debed00bfbe33846f99c4540f79b>`_ from Docker Hub.