From cc9c2c31e54c5bd8c1f9f17a6dae17dce4d9af7a Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 23 Jan 2026 14:34:17 +0300 Subject: [PATCH] Update ROCm Linux and OpeinVINO Torch to 2.10 --- installer.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/installer.py b/installer.py index 6369c76df..d3a040406 100644 --- a/installer.py +++ b/installer.py @@ -804,7 +804,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: - if rocm.version is None or float(rocm.version) >= 6.4: # assume the latest if version check fails + 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') @@ -854,9 +858,9 @@ def install_openvino(): #check_python(supported_minors=[10, 11, 12, 13], reason='OpenVINO backend requires a Python version between 3.10 and 3.13') if sys.platform == 'darwin': - torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.9.1 torchvision==0.24.1') + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.10.0 torchvision==0.25.0') else: - torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.9.1+cpu torchvision==0.24.1 --index-url https://download.pytorch.org/whl/cpu') + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.10.0+cpu torchvision==0.25.0 --index-url https://download.pytorch.org/whl/cpu') if not (args.skip_all or args.skip_requirements): install(os.environ.get('OPENVINO_COMMAND', 'openvino==2025.3.0'), 'openvino')