Skip to content

[Installation][build][docker]: rocm Dockerfile pinned to stale python torch nightly wheel builds #12066

@ghost

Description

How you are installing vllm

vllm/Dockerfile.rocm

Lines 48 to 58 in 0794e74

# Install torch == 2.6.0 on ROCm
RUN --mount=type=cache,target=/root/.cache/pip \
case "$(ls /opt | grep -Po 'rocm-[0-9]\.[0-9]')" in \
*"rocm-6.2"*) \
python3 -m pip uninstall -y torch torchvision \
&& python3 -m pip install --pre \
torch==2.6.0.dev20241113+rocm6.2 \
'setuptools-scm>=8' \
torchvision==0.20.0.dev20241113+rocm6.2 \
--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2;; \
*) ;; esac

Python packages for torch==2.6.0.dev20241113+rocm6.2 and torchvision==0.20.0.dev20241113+rocm6.2 are no longer available due to them being outside of the build retention window

Wheel Index:
https://download.pytorch.org/whl/nightly/torch/

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

Metadata

Metadata

Assignees

No one assigned

    Labels

    installationInstallation problemsrocmRelated to AMD ROCm

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions