Revert "Switch to ROCm 6.4 + Python 3.11 + PyTorch 2.9.1"

This reverts commit eb6a39d292.
2026-04-06 18:05:43 -04:00
parent eb6a39d292
commit 66445fad84
2 changed files with 12 additions and 22 deletions
--- a/requirements-rocm-init.txt
+++ b/requirements-rocm-init.txt
@@ -1,5 +1,5 @@
--index-url https://download.pytorch.org/whl/rocm6.4
-torch==2.9.1
-torchaudio==2.9.1
-torchvision==0.24.1
-pytorch_triton_rocm==3.5.1
+--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