From 1a7b88ab6450b2cf4e80843fdbff468dcb8aa73b Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 17 Oct 2024 23:14:00 +0300 Subject: [PATCH] Update the default ROCm version --- installer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/installer.py b/installer.py index 5db1f6205..0e08f06f5 100644 --- a/installer.py +++ b/installer.py @@ -566,8 +566,8 @@ def install_rocm_zluda(): log.info('Using CPU-only torch') torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision') else: - if rocm.version is None or float(rocm.version) > 6.1: # assume the latest if version check fails - torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm6.1') + if rocm.version is None or float(rocm.version) > 6.2: # assume the latest if version check fails + torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision --index-url https://download.pytorch.org/whl/rocm6.2') elif float(rocm.version) < 5.5: # oldest supported version is 5.5 log.warning(f"ROCm: unsupported version={rocm.version}") log.warning("ROCm: minimum supported version=5.5")