mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 09:35:50 +08:00
[Doc] Update rocm.inc.md (#15917)
Signed-off-by: chun37 <chun.jb.37@gmail.com>
This commit is contained in:
parent
274d8e8818
commit
c920e01242
@ -31,7 +31,7 @@ Currently, there are no pre-built ROCm wheels.
|
|||||||
```console
|
```console
|
||||||
# Install PyTorch
|
# Install PyTorch
|
||||||
$ pip uninstall torch -y
|
$ 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)
|
1. Install [Triton flash attention for ROCm](https://github.com/ROCm/triton)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user