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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user