20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: chun37 <chun.jb.37@gmail.com>
This commit is contained in:
chun 2025-04-02 15:38:26 +09:00 committed by GitHub
parent 274d8e8818
commit c920e01242
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/getting_started/installation/gpu/rocm.inc.md
View File

@ -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)