From dc7a3cf769d2c5f07401c1a497e28c93ec9933ef Mon Sep 17 00:00:00 2001 From: scott Date: Sun, 5 Apr 2026 11:06:39 -0400 Subject: [PATCH] Upgrade to ROCm 7.2 and PyTorch 2.11.0 - Update torch/torchaudio to 2.11.0 with ROCm 7.2 wheel index - Drop torchvision (unused for TTS) and pytorch_triton_rocm (bundled in 2.11) - Update HSA_OVERRIDE_GFX_VERSION docs; RX 7000+ natively supported in ROCm 7.2 Co-Authored-By: Claude Sonnet 4.6 --- docker-compose.yml | 7 ++++--- requirements-rocm-init.txt | 8 +++----- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index dbb8a46..f9952b9 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -24,11 +24,12 @@ services: - hf_cache:/app/hf_cache environment: - HF_HUB_ENABLE_HF_TRANSFER=1 - # Set your GPU architecture: + # HSA_OVERRIDE_GFX_VERSION is only needed for GPUs not natively supported + # by the installed ROCm version. ROCm 7.2 supports RX 7000 series and newer + # natively. Older cards may still need an override: # 10.3.0 = RX 5000/6000 series - # 11.0.0 = RX 7000 series # 9.0.6 = Vega - - HSA_OVERRIDE_GFX_VERSION=10.3.0 + # - HSA_OVERRIDE_GFX_VERSION=10.3.0 # - HF_TOKEN=your_token_here volumes: diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt index 44a44f9..d19275f 100644 --- a/requirements-rocm-init.txt +++ b/requirements-rocm-init.txt @@ -1,5 +1,3 @@ ---index-url https://download.pytorch.org/whl/rocm6.1 -torch==2.5.1 -torchaudio==2.5.1 -torchvision==0.20.1 -pytorch_triton_rocm==3.1.0 +--index-url https://download.pytorch.org/whl/rocm7.2 +torch==2.11.0 +torchaudio==2.11.0