Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
ced36cd8
Unverified
Commit
ced36cd8
authored
Jul 26, 2024
by
Woosuk Kwon
Committed by
GitHub
Jul 26, 2024
Browse files
[ROCm] Upgrade PyTorch nightly version (#6845)
parent
969d0322
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
Dockerfile.rocm
Dockerfile.rocm
+2
-2
docs/source/getting_started/amd-installation.rst
docs/source/getting_started/amd-installation.rst
+1
-1
No files found.
Dockerfile.rocm
View file @
ced36cd8
...
...
@@ -55,8 +55,8 @@ RUN case "$(ls /opt | grep -Po 'rocm-[0-9]\.[0-9]')" in \
*"rocm-6.1"*) \
python3 -m pip uninstall -y torch torchvision \
&& python3 -m pip install --no-cache-dir --pre \
torch==2.5.0.dev202407
10
\
torchvision==0.20.0.dev202407
10
\
torch==2.5.0.dev202407
26
\
torchvision==0.20.0.dev202407
26
\
--index-url https://download.pytorch.org/whl/nightly/rocm6.1;; \
*) ;; esac
...
...
docs/source/getting_started/amd-installation.rst
View file @
ced36cd8
...
...
@@ -117,7 +117,7 @@ Alternatively, wheels intended for vLLM use can be accessed under the releases.
$ # Install PyTorch
$ pip uninstall torch -y
$ pip install --no-cache-dir --pre torch==2.5.0.dev202407
10
--index-url https://download.pytorch.org/whl/nightly/rocm6.1
$ pip install --no-cache-dir --pre torch==2.5.0.dev202407
26
--index-url https://download.pytorch.org/whl/nightly/rocm6.1
$ # Build & install AMD SMI
$ pip install /opt/rocm/share/amd_smi
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment