Skip to content

GitLab

Unverified Commit c920e012 authored by chun's avatar chun Committed by GitHub
Browse files

[Doc] Update rocm.inc.md (#15917) 


Signed-off-by: default avatarchun37 <chun.jb.37@gmail.com>
parent 274d8e88
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/getting_started/installation/gpu/rocm.inc.md
View file @ c920e012
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment