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

[ghost]

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.

[ghost]

In Process:
Building and verifying with latest nightlies for rocm6.2

torch==2.6.0.dev20241122+rocm6.2
torchvision==0.20.0.dev20241123+rocm6.2

[hongxiayang]

@cob-web-corner Please test this one since the PyTorch stable release is already on ROCm 6.2 :

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2

We will check to update the Dockerfile.

[ghost]

@hongxiayang Thank you, not a usual AMD+PyTorch user

Confirmed working on MI300X, MI300A and MI250