From 8206b77f34e2509c9743a39edc9e9676c4af381f Mon Sep 17 00:00:00 2001 From: Rohan Potdar Date: Tue, 20 Oct 2026 18:39:29 +0000 Subject: [PATCH] [ROCm][CI] Bump the ROCm 10.0 image to torch 2.14 Move TheRock's torch/torchvision to 2.14.0/0.29.0 (+rocm10.0.0), matching CUDA/CPU/XPU, and update the install docs. Co-authored-by: Claude Signed-off-by: Rohan Potdar --- docker/Dockerfile.rocm_base | 4 ++-- docs/getting_started/installation/gpu.rocm.inc.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index 76bb1bc00aeb..22cb6e456a16 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -1,7 +1,7 @@ ARG BASE_IMAGE=ubuntu:22.04 ARG PYTHON_VERSION=3.12 -ARG TORCH_VERSION=2.13.0+rocm10.0.0 -ARG TORCHVISION_VERSION=0.28.0+rocm10.0.0 +ARG TORCH_VERSION=2.14.0+rocm10.0.0 +ARG TORCHVISION_VERSION=0.29.0+rocm10.0.0 ARG TORCHAUDIO_VERSION=2.11.0+rocm10.0.0 ARG TRITON_BRANCH="669b31ac" # release/internal/3.8.x as of 09/29/2026 ARG TRITON_VERSION=3.8.0+git669b31ac diff --git a/docs/getting_started/installation/gpu.rocm.inc.md b/docs/getting_started/installation/gpu.rocm.inc.md index d8debbab3ea3..4018a7e5993c 100644 --- a/docs/getting_started/installation/gpu.rocm.inc.md +++ b/docs/getting_started/installation/gpu.rocm.inc.md @@ -247,8 +247,8 @@ You can find more information about vLLM's wheels in pip install --index-url ${ROCM_INDEX} \ "rocm[libraries,devel,device-${GPU_ARCH}]==10.0.0" \ - "torch[device-${GPU_ARCH}]==2.12.0+rocm10.0.0" \ - "torchvision[device-${GPU_ARCH}]==0.27.0+rocm10.0.0" + "torch[device-${GPU_ARCH}]==2.14.0+rocm10.0.0" \ + "torchvision[device-${GPU_ARCH}]==0.29.0+rocm10.0.0" rocm-sdk init # Point builds at the pip-installed ROCm SDK