wassname/vllm
1
0 Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-10-03 13:02:20 +08:00
Code Issues 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 authored and GitHub committed 2025-01-18 11:15:53 +08:00
1 parent 2b83503227
commit c09503ddd6
2 files changed
+4 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

docs/source/getting_started/installation/gpu/rocm.inc.md
+1 -1
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