diff --git a/CHANGELOG.md b/CHANGELOG.md index 6b5ec8b23..49b826ad1 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -2,7 +2,6 @@ ## Todo -- **rocm/linux**: update to `torch==2.10.0` - **rocm/windows**: update to `torch==2.10.0` ## Update for 2026-01-27 @@ -23,6 +22,7 @@ *note*: may cause slow first startup/generate **cuda**: update to `torch==2.10.0` **xpu**: update to `torch==2.10.0` + **rocm/linux**: update to `torch==2.10.0` **openvino**: update to `torch==2.10.0` and `openvino==2025.4.1` - relocate all json data files to `data/` folder existing data files are auto-migrated on startup diff --git a/installer.py b/installer.py index 74c7a8f46..f711a84e4 100644 --- a/installer.py +++ b/installer.py @@ -820,13 +820,11 @@ def install_rocm_zluda(): else: # oldest rocm version on nightly is 7.0 torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.0') else: - # Torch 2.10 + ROCm has AoT and performance regressions - #if rocm.version is None or float(rocm.version) >= 7.1: # assume the latest if version check fails - # torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.10.0+rocm7.1 torchvision==0.25.0+rocm7.1 --index-url https://download.pytorch.org/whl/rocm7.1') - #elif rocm.version == "7.0": # assume the latest if version check fails - # torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.10.0+rocm7.0 torchvision==0.25.0+rocm7.0 --index-url https://download.pytorch.org/whl/rocm7.0') - #elif rocm.version == "6.4": - if rocm.version is None or float(rocm.version) >= 6.4: + if rocm.version is None or float(rocm.version) >= 7.1: # assume the latest if version check fails + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.10.0+rocm7.1 torchvision==0.25.0+rocm7.1 --index-url https://download.pytorch.org/whl/rocm7.1') + elif rocm.version == "7.0": # assume the latest if version check fails + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.10.0+rocm7.0 torchvision==0.25.0+rocm7.0 --index-url https://download.pytorch.org/whl/rocm7.0') + elif rocm.version == "6.4": torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.9.1+rocm6.4 torchvision==0.24.1+rocm6.4 --index-url https://download.pytorch.org/whl/rocm6.4') elif rocm.version == "6.3": torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.9.1+rocm6.3 torchvision==0.24.1+rocm6.3 --index-url https://download.pytorch.org/whl/rocm6.3')