diff --git a/requirements/rocm-build.txt b/requirements/rocm-build.txt index a0731c51d46b..6af78da4993d 100644 --- a/requirements/rocm-build.txt +++ b/requirements/rocm-build.txt @@ -1,10 +1,10 @@ # Common dependencies -r common.txt ---extra-index-url https://download.pytorch.org/whl/rocm6.2 -torch==2.5.1 -torchvision==0.20.1 -torchaudio==2.5.1 +--extra-index-url https://download.pytorch.org/whl/rocm6.2.4 +torch==2.6.0 +torchvision==0.21.0 +torchaudio==2.6.0 cmake>=3.26 packaging