Upgrade to ROCm 7.2 and PyTorch 2.11.0
Some checks failed
Build ROCm Image / build (push) Failing after 7m25s

- 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 <noreply@anthropic.com>
This commit is contained in:
2026-04-05 11:06:39 -04:00
parent d7247d31fe
commit dc7a3cf769
2 changed files with 7 additions and 8 deletions

View File

@@ -24,11 +24,12 @@ services:
- hf_cache:/app/hf_cache - hf_cache:/app/hf_cache
environment: environment:
- HF_HUB_ENABLE_HF_TRANSFER=1 - 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 # 10.3.0 = RX 5000/6000 series
# 11.0.0 = RX 7000 series
# 9.0.6 = Vega # 9.0.6 = Vega
- HSA_OVERRIDE_GFX_VERSION=10.3.0 # - HSA_OVERRIDE_GFX_VERSION=10.3.0
# - HF_TOKEN=your_token_here # - HF_TOKEN=your_token_here
volumes: volumes:

View File

@@ -1,5 +1,3 @@
--index-url https://download.pytorch.org/whl/rocm6.1 --index-url https://download.pytorch.org/whl/rocm7.2
torch==2.5.1 torch==2.11.0
torchaudio==2.5.1 torchaudio==2.11.0
torchvision==0.20.1
pytorch_triton_rocm==3.1.0