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

[AMD][CI/Build][Bugfix] use pytorch stale wheel (#12172)

Browse Source 
Signed-off-by: hongxyan <hongxyan@amd.com>
This commit is contained in:
Hongxia Yang
2025-01-17 22:15:53 -05:00
committed by GitHub
parent 2b83503227
commit c09503ddd6
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -70,7 +70,7 @@ Currently, there are no pre-built ROCm wheels.
# Install PyTorch
$ pip uninstall torch -y
$ pip install --no-cache-dir --pre torch==2.6.0.dev20241024 --index-url https://download.pytorch.org/whl/nightly/rocm6.2
$ pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/rocm6.2
# Build & install AMD SMI
$ pip install /opt/rocm/share/amd_smi