Upgrade to ROCm 7.2, Python 3.11, PyTorch 2.11.0

- Base image: rocm/dev-ubuntu-22.04:6.1 → 7.2
- Python 3.10 → 3.11 via deadsnakes PPA
- torch/torchaudio: 2.5.1 → 2.11.0
- torchvision: 0.20.1 → 0.26.0
- pytorch_triton_rocm: 3.1.0 → 3.3.0
- transformers: 4.46.3 → >=4.50.0
- diffusers: 0.29.0 → >=0.32.0
- safetensors: >=0.4.1 → >=0.4.5
- config: temperature 0.8→0.9, seed 0→1960

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-04-06 17:09:56 -04:00
parent 0fac076de1
commit fe3c77ff4f
4 changed files with 27 additions and 17 deletions

View File

@@ -1,5 +1,5 @@
--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
torchvision==0.26.0
pytorch_triton_rocm==3.3.0