From 101983faefb2086cfdce8a741dc045d16cf4cf66 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 12 Dec 2025 12:54:46 +0300 Subject: [PATCH] Update ROCm nightlies to ROCm 7.1 --- installer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/installer.py b/installer.py index fecdb8db6..3503e14a2 100644 --- a/installer.py +++ b/installer.py @@ -781,10 +781,10 @@ def install_rocm_zluda(): else: #check_python(supported_minors=[10, 11, 12, 13, 14], reason='ROCm backend requires a Python version between 3.10 and 3.13') if args.use_nightly: - if rocm.version is None or float(rocm.version) >= 7.0: # 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', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1') + 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: # oldest rocm version on nightly is 6.4 - torch_command = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4') else: if rocm.version is None or float(rocm.version) >= 6.4: # assume the latest if version check fails 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')