From f15cdcf04906300cc78412d2825c3ad8bf9a676b Mon Sep 17 00:00:00 2001 From: scott Date: Sun, 5 Apr 2026 12:35:58 -0400 Subject: [PATCH] Pin base image to rocm/dev-ubuntu-22.04:7.2, restore torch 2.11.0 Using :latest was pulling a ROCm 6.x image whose MIOpen was incompatible with our ROCm 7.2 PyTorch wheels. Pinning to the 7.2 tag gives matching MIOpen libraries and should resolve the workspace/fallback performance issue. Co-Authored-By: Claude Sonnet 4.6 --- Dockerfile.rocm | 2 +- requirements-rocm-init.txt | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/Dockerfile.rocm b/Dockerfile.rocm index f94cc54..9411eef 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,4 +1,4 @@ -FROM rocm/dev-ubuntu-22.04:latest +FROM rocm/dev-ubuntu-22.04:7.2 ENV DEBIAN_FRONTEND=noninteractive \ PYTHONDONTWRITEBYTECODE=1 \ diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt index 7ce9157..d19275f 100644 --- a/requirements-rocm-init.txt +++ b/requirements-rocm-init.txt @@ -1,3 +1,3 @@ ---index-url https://download.pytorch.org/whl/rocm6.1 -torch==2.5.1 -torchaudio==2.5.1 +--index-url https://download.pytorch.org/whl/rocm7.2 +torch==2.11.0 +torchaudio==2.11.0