Pin base image to rocm/dev-ubuntu-22.04:7.2, restore torch 2.11.0
All checks were successful
Build ROCm Image / build (push) Successful in 16m4s
All checks were successful
Build ROCm Image / build (push) Successful in 16m4s
Using :latest was pulling a ROCm 6.x image whose MIOpen was incompatible with our ROCm 7.2 PyTorch wheels. Pinning to the 7.2 tag gives matching MIOpen libraries and should resolve the workspace/fallback performance issue. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -1,3 +1,3 @@
|
||||
--index-url https://download.pytorch.org/whl/rocm6.1
|
||||
torch==2.5.1
|
||||
torchaudio==2.5.1
|
||||
--index-url https://download.pytorch.org/whl/rocm7.2
|
||||
torch==2.11.0
|
||||
torchaudio==2.11.0
|
||||
|
||||
Reference in New Issue
Block a user