Re-update ROCm / Linux to Torch 2.10

This commit is contained in:
Disty0
2026-01-27 16:07:20 +03:00
parent 7ce0cc2520
commit 727fd92d76
2 changed files with 6 additions and 8 deletions
+1 -1
View File
@@ -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
+5 -7
View File
@@ -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')