diff --git a/requirements/rocm-build.txt b/requirements/rocm-build.txt index 01a71c2da..6f96c7d55 100644 --- a/requirements/rocm-build.txt +++ b/requirements/rocm-build.txt @@ -1,7 +1,7 @@ # Common dependencies -r common.txt ---extra-index-url https://download.pytorch.org/whl/test/rocm7.0 +--extra-index-url https://download.pytorch.org/whl/rocm7.1 torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 @@ -12,5 +12,5 @@ setuptools>=77.0.3,<80.0.0 setuptools-scm>=8 wheel jinja2>=3.1.6 -amdsmi==6.4.3 +amdsmi==7.0.2 timm>=1.0.17