From e0eb7c36df0699b690ba93b95e01cc72bb89c920 Mon Sep 17 00:00:00 2001 From: scott Date: Mon, 6 Apr 2026 18:04:59 -0400 Subject: [PATCH] Switch to ROCm 6.4 + Python 3.11 + PyTorch 2.9.1 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Base image: rocm/dev-ubuntu-22.04:6.4 - Python 3.10 → 3.11 via deadsnakes PPA - torch/torchaudio: 2.5.1 → 2.9.1 (rocm6.4 wheel index) - torchvision: 0.20.1 → 0.24.1 - pytorch_triton_rocm: 3.1.0 → 3.5.1 Co-Authored-By: Claude Sonnet 4.6 --- Dockerfile.rocm | 2 +- requirements-rocm-init.txt | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 49db5e9..43812c4 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,4 +1,4 @@ -FROM rocm/dev-ubuntu-22.04:7.2 +FROM rocm/dev-ubuntu-22.04:6.4 ENV DEBIAN_FRONTEND=noninteractive \ PYTHONDONTWRITEBYTECODE=1 \ diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt index 8627435..ce4fabe 100644 --- a/requirements-rocm-init.txt +++ b/requirements-rocm-init.txt @@ -1,5 +1,5 @@ ---index-url https://download.pytorch.org/whl/rocm7.2 -torch==2.11.0 -torchaudio==2.11.0 -torchvision==0.26.0 -pytorch_triton_rocm==3.3.0 +--index-url https://download.pytorch.org/whl/rocm6.4 +torch==2.9.1 +torchaudio==2.9.1 +torchvision==0.24.1 +pytorch_triton_rocm==3.5.1