From 100ca6622165ef6a4c791b02c0caaa2887ca41b7 Mon Sep 17 00:00:00 2001 From: chun Date: Wed, 2 Apr 2025 11:01:02 +0900 Subject: [PATCH] Update rocm.inc.md Signed-off-by: chun37 --- docs/source/getting_started/installation/gpu/rocm.inc.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/getting_started/installation/gpu/rocm.inc.md b/docs/source/getting_started/installation/gpu/rocm.inc.md index eae7a2358510..21c8d7d01ade 100644 --- a/docs/source/getting_started/installation/gpu/rocm.inc.md +++ b/docs/source/getting_started/installation/gpu/rocm.inc.md @@ -31,7 +31,7 @@ Currently, there are no pre-built ROCm wheels. ```console # Install PyTorch $ pip uninstall torch -y - $ pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/rocm6.3 + $ pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/nightly/rocm6.3 ``` 1. Install [Triton flash attention for ROCm](https://github.com/ROCm/triton)