github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-01-09 22:58:17 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix link to ROCm PyT docker image (#5803)

Browse Source
This commit is contained in:
peterjunpark
2025-12-19 15:47:55 -05:00
committed by GitHub
parent 7455fe57b8
commit 48d8fe139b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/how-to/rocm-for-ai/inference-optimization/model-acceleration-libraries.rst
View File

@@ -33,8 +33,8 @@ Before installing Flash Attention 2, ensure the following are available:
* Triton * Triton
These can be installed by following the official These can be installed by following the official
`PyTorch installation guide <https://pytorch.org/get-started/locally/>`_. Alternatively, for a simpler setup, you can use a preconfigured `PyTorch installation guide <https://pytorch.org/get-started/locally/>`_. Alternatively, for a simpler setup, you can use a preconfigured
`ROCm PyTorch Docker image <https://github.com/ROCm/rocm-docker-images>`_, which already includes the required libraries. :ref:`ROCm PyTorch Docker image <using-docker-with-pytorch-pre-installed>`, which already includes the required libraries.
Installing Flash Attention 2 Installing Flash Attention 2
---------------------------- ----------------------------